Permalink
Browse files
Fix crop layer lint errors
- Loading branch information...
Showing
with
2 additions
and
0 deletions.
-
+2
−0
src/caffe/layers/crop_layer.cu
|
|
@@ -54,6 +54,7 @@ void CropLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom, |
|
|
const Dtype* bottom_data = bottom[0]->gpu_data();
|
|
|
Dtype* top_data = top[0]->mutable_gpu_data();
|
|
|
int n = top[0]->count();
|
|
|
+ // NOLINT_NEXT_LINE(whitespace/operators)
|
|
|
crop_kernel_forward<<<CAFFE_GET_BLOCKS(n), CAFFE_CUDA_NUM_THREADS>>>(n,
|
|
|
bottom[0]->num_axes(),
|
|
|
src_strides_.gpu_data(),
|
|
|
@@ -71,6 +72,7 @@ void CropLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top, |
|
|
|
|
|
if (propagate_down[0]) {
|
|
|
caffe_gpu_set(bottom[0]->count(), static_cast<Dtype>(0), bottom_diff);
|
|
|
+ // NOLINT_NEXT_LINE(whitespace/operators)
|
|
|
crop_kernel_backward<<<CAFFE_GET_BLOCKS(n), CAFFE_CUDA_NUM_THREADS>>>(n,
|
|
|
bottom[0]->num_axes(),
|
|
|
src_strides_.gpu_data(),
|
|
|
|
0 comments on commit
cd1696d