fix bug of DepthwiseConvTransposeGradKernel (#31762)

5 years ago · a45c8ca69d
parent 25fc2a1fdb
commit a45c8ca69d
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/operators/conv_transpose_op.h
+++ b/paddle/fluid/operators/conv_transpose_op.h
@ -682,9 +682,9 @@ class DepthwiseConvTransposeGradKernel : public framework::OpKernel<T> {
    if (input_grad) {
      math::DepthwiseConvFunctor<DeviceContext, T> depthwiseConv;
      depthwiseConv(
-          dev_ctx, *output_grad, filter, strides, paddings,
+          dev_ctx, *output_grad, filter, strides,
          std::vector<int>{paddings[0], paddings[2], paddings[1], paddings[3]},
-          input_grad, data_layout);
+          dilations, input_grad, data_layout);
    }

    if (filter_grad) {