提交 aec10c8a 编写于 作者: J jiangjiajun

fix bug for conv2d_transpose

上级 85e3cf41
......@@ -666,9 +666,9 @@ class PaddleEmitter(object):
k_shape = self.tensor_shape_to_list(
kernel.get_attr("_output_shapes"))[0]
k_num, channel, k_h, k_w = k_shape
channel, k_num, k_h, k_w = k_shape
if node.data_format == "NHWC":
k_h, k_w, channel, k_num = k_shape
k_h, k_w, k_num, channel = k_shape
if strides[0] > k_h or strides[1] > k_w:
raise Exception(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册