未验证 提交 45a91158 编写于 作者: W Weilong Wu 提交者: GitHub

[Eager] fix unpool and yaml config (#45544)

上级 cf9d651b
......@@ -2715,7 +2715,6 @@
kernel:
func: unpool3d_grad
data_type: x
no_need_buffer : x
- backward_api: unpool_grad
forward: unpool (Tensor x, Tensor indices, int[] ksize, int[] strides, int[] padding, int[] output_size, str data_format) -> Tensor(out)
......@@ -2727,4 +2726,3 @@
kernel:
func: unpool_grad
data_type: x
no_need_buffer : x
......@@ -878,7 +878,7 @@ def max_unpool2d(x,
if in_dygraph_mode():
output = _C_ops.unpool(x, indices, kernel_size, stride, padding,
output_size, data_format)
return output
elif in_dynamic_mode():
output = _legacy_C_ops.unpool(x, indices, 'unpooling_type', 'max',
'ksize', kernel_size, 'strides', stride,
......@@ -995,6 +995,7 @@ def max_unpool3d(x,
if in_dygraph_mode():
output = _C_ops.unpool3d(x, indices, kernel_size, stride, padding,
output_size, data_format)
return output
elif in_dynamic_mode():
output = _legacy_C_ops.unpool3d(x, indices, 'unpooling_type', 'max',
'ksize', kernel_size, 'strides', stride,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册