Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 a45c8ca6 编写于 作者: O Ouyang Chao 提交者： GitHub
浏览文件
操作

fix bug of DepthwiseConvTransposeGradKernel (#31762)

上级 25fc2a1f
显示空白变更内容
内联 并排
Showing with 2 addition and 2 deletion
paddle/fluid/operators/conv_transpose_op.h
浏览文件 @ a45c8ca6
...@@ -682,9 +682,9 @@ class DepthwiseConvTransposeGradKernel : public framework::OpKernel<T> { ...@@ -682,9 +682,9 @@ class DepthwiseConvTransposeGradKernel : public framework::OpKernel<T> {
if (input_grad) { if (input_grad) {
math::DepthwiseConvFunctor<DeviceContext, T> depthwiseConv; math::DepthwiseConvFunctor<DeviceContext, T> depthwiseConv;
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]}, std::vector<int>{paddings[0], paddings[2], paddings[1], paddings[3]},
input_grad, data_layout); dilations, input_grad, data_layout);
} }
if (filter_grad) { if (filter_grad) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或