Instantiate caffe_cpu_sign for float and double

This commit is contained in:
Kai Li 2014-02-25 19:21:18 +08:00
Родитель 348a338e7f
Коммит f634899f44
1 изменённых файлов: 6 добавлений и 0 удалений

Просмотреть файл

@ -410,4 +410,10 @@ void caffe_gpu_asum<double>(const int n, const double* x, double* y) {
CUBLAS_CHECK(cublasDasum(Caffe::cublas_handle(), n, x, 1, y));
}
template <>
void caffe_cpu_sign<float>(const int n, const float* x, float* y);
template <>
void caffe_cpu_sign<double>(const int n, const double* x, double* y);
} // namespace caffe