diff options
Diffstat (limited to 'src/caffe/layers/sigmoid_layer.cu')
-rw-r--r-- | src/caffe/layers/sigmoid_layer.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/sigmoid_layer.cu b/src/caffe/layers/sigmoid_layer.cu index 6879ff7f..e1af0657 100644 --- a/src/caffe/layers/sigmoid_layer.cu +++ b/src/caffe/layers/sigmoid_layer.cu @@ -56,7 +56,7 @@ void SigmoidLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top, } } -INSTANTIATE_CLASS(SigmoidLayer); +INSTANTIATE_LAYER_GPU_FUNCS(SigmoidLayer); } // namespace caffe |