diff options
Diffstat (limited to 'src/caffe/layers/slice_layer.cu')
-rw-r--r-- | src/caffe/layers/slice_layer.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/slice_layer.cu b/src/caffe/layers/slice_layer.cu index 65784080..b5c5e615 100644 --- a/src/caffe/layers/slice_layer.cu +++ b/src/caffe/layers/slice_layer.cu @@ -63,6 +63,6 @@ void SliceLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top, } // slice_dim_ is guaranteed to be 0 or 1 by SetUp. } -INSTANTIATE_CLASS(SliceLayer); +INSTANTIATE_LAYER_GPU_FUNCS(SliceLayer); } // namespace caffe |