diff --git a/paddle/operators/conv2dtranspose_op.h b/paddle/operators/conv2dtranspose_op.h index 293b7ce9ba9454e50ca14eb947de293de0bb56ff..8c70b3dcec1e26ab3d8a42d88040764c643b5ae6 100644 --- a/paddle/operators/conv2dtranspose_op.h +++ b/paddle/operators/conv2dtranspose_op.h @@ -120,7 +120,7 @@ class GemmConv2DTransposeKernel : public framework::OpKernel { math::matmul(context.device_context(), filter, true, input_batch, false, T(1.0), &col_matrix, T(0.0)); col2im(context.device_context(), output_batch, col, strides[0], - strides[1], 0, 0); + strides[1], 0, 0, 0, 0); } } }; @@ -206,7 +206,7 @@ class GemmConv2DTransposeGradKernel : public framework::OpKernel { // im2col: dy from (c, o_h, o_w) -> (c * k_h * k_w, h * w) im2col(context.device_context(), output_grad_batch, col, strides[0], - strides[1], paddings[0], paddings[1]); + strides[1], paddings[0], paddings[0], paddings[1], paddings[1]); // gemm: dx = filter * dy // (m, c * k_h * k_w) * (c * k_h * k_w, h * w) -> (m, c, h) @@ -238,7 +238,7 @@ class GemmConv2DTransposeGradKernel : public framework::OpKernel { // im2col: (c * h * w, k_h * k_w) im2col(context.device_context(), output_grad_batch, col, strides[0], - strides[1], paddings[0], paddings[1]); + strides[1], paddings[0], paddings[0], paddings[1], paddings[1]); // gemm: d_filter = x * y_grad^T // (m, c * h * w) * (k_h * k_w, c * h * w) -> (m, c, h) diff --git a/paddle/operators/math/im2col.h b/paddle/operators/math/im2col.h index a1cb956c51ebfa80892745b88a6d85a0f09202d6..c736d4fa523c2af3e3dd7a11114d7f84021bc5c1 100644 --- a/paddle/operators/math/im2col.h +++ b/paddle/operators/math/im2col.h @@ -75,8 +75,7 @@ class Im2ColFunctor { void operator()(const platform::DeviceContext& context, const framework::Tensor& im, framework::Tensor& col, int stride_height, int stride_width, int padding_up, - int padding_down, int padding_left = 0, - int padding_right = 0); + int padding_down, int padding_left, int padding_right); }; template @@ -85,7 +84,7 @@ class Col2ImFunctor { void operator()(const platform::DeviceContext& context, framework::Tensor& im, const framework::Tensor& col, int stride_height, int stride_width, int padding_up, int padding_down, - int padding_left = 0, int padding_right = 0); + int padding_left, int padding_right); }; } // namespace math