зеркало из https://github.com/microsoft/caffe.git
Instantiate caffe_cpu_sign for float and double
This commit is contained in:
Родитель
348a338e7f
Коммит
f634899f44
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче