diff --git a/paddle/function/CropOpGpu.cu b/paddle/function/CropOpGpu.cu index 37ce6de0647e5e06a231710b5a53089533de2407..786eb268d45aadee0c1f6fcbbafc23173cf0bc77 100644 --- a/paddle/function/CropOpGpu.cu +++ b/paddle/function/CropOpGpu.cu @@ -36,10 +36,11 @@ __global__ void KeCrop(real* outputs, const real* inputs, template <> void Crop(real* outputs, const real* inputs, - const TensorShape inShape, - const TensorShape outShape, + const TensorShape inShape, + const TensorShape outShape, const FuncConfig& conf) { - std::vector crop_corner = conf.get>("crop_corner"); + std::vector crop_corner = + conf.get>("crop_corner"); int cropC = crop_corner[1]; int cropH = crop_corner[2]; int cropW = crop_corner[3]; @@ -74,7 +75,8 @@ __global__ void KeCropDiff(const real* inGrad, real* outGrad, const int c = (idx / inW / inH) % inC; const int n = idx / inW / inH / inC; - const int off = ((n * outC + c + cropC) * outH + h + cropH) * outW + cropW + w; + const int off = + ((n * outC + c + cropC) * outH + h + cropH) * outW + cropW + w; outGrad[off] += inGrad[idx]; } @@ -86,7 +88,8 @@ void CropGrad(const real* inGrad, const TensorShape inShape, const TensorShape outShape, const FuncConfig& conf) { - std::vector crop_corner = conf.get>("crop_corner"); + std::vector crop_corner = + conf.get>("crop_corner"); int cropC = crop_corner[1]; int cropH = crop_corner[2]; int cropW = crop_corner[3];