diff options
Diffstat (limited to 'src/caffe/layers/im2col_layer.cu')
-rw-r--r-- | src/caffe/layers/im2col_layer.cu | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/caffe/layers/im2col_layer.cu b/src/caffe/layers/im2col_layer.cu index 6b4c7010..8df061d8 100644 --- a/src/caffe/layers/im2col_layer.cu +++ b/src/caffe/layers/im2col_layer.cu @@ -8,7 +8,7 @@ namespace caffe { template <typename Dtype> -Dtype Im2colLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, +void Im2colLayer<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(); @@ -17,7 +17,6 @@ Dtype Im2colLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, width_, kernel_h_, kernel_w_, pad_h_, pad_w_, stride_h_, stride_w_, top_data + (*top)[0]->offset(n)); } - return Dtype(0.); } template <typename Dtype> |