diff options
Diffstat (limited to 'src/caffe/layers/inner_product_layer.cu')
-rw-r--r-- | src/caffe/layers/inner_product_layer.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/inner_product_layer.cu b/src/caffe/layers/inner_product_layer.cu index 2164b4d4..a9e1784a 100644 --- a/src/caffe/layers/inner_product_layer.cu +++ b/src/caffe/layers/inner_product_layer.cu @@ -51,6 +51,6 @@ void InnerProductLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top, } } -INSTANTIATE_CLASS(InnerProductLayer); +INSTANTIATE_LAYER_GPU_FUNCS(InnerProductLayer); } // namespace caffe |