diff options
Diffstat (limited to 'src/caffe/layers/pooling_layer.cu')
-rw-r--r-- | src/caffe/layers/pooling_layer.cu | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/caffe/layers/pooling_layer.cu b/src/caffe/layers/pooling_layer.cu index 58f1997c..e64128b8 100644 --- a/src/caffe/layers/pooling_layer.cu +++ b/src/caffe/layers/pooling_layer.cu @@ -151,7 +151,7 @@ __global__ void StoPoolForwardTest(const int nthreads, template <typename Dtype> -Dtype PoolingLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, +void PoolingLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, vector<Blob<Dtype>*>* top) { const Dtype* bottom_data = bottom[0]->gpu_data(); Dtype* top_data = (*top)[0]->mutable_gpu_data(); @@ -206,7 +206,6 @@ Dtype PoolingLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, LOG(FATAL) << "Unknown pooling method."; } CUDA_POST_KERNEL_CHECK; - return Dtype(0.); } |