未验证 提交 a45c8ca6 编写于 作者: O Ouyang Chao 提交者: GitHub

fix bug of DepthwiseConvTransposeGradKernel (#31762)

上级 25fc2a1f
......@@ -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) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册