From cafb4c66aa8d0fe3e113cec22159566200843f4b Mon Sep 17 00:00:00 2001 From: Channingss Date: Fri, 7 Aug 2020 02:28:18 +0000 Subject: [PATCH] update --- x2paddle/op_mapper/paddle2onnx/opset9/opset.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/x2paddle/op_mapper/paddle2onnx/opset9/opset.py b/x2paddle/op_mapper/paddle2onnx/opset9/opset.py index 742515d..89c6525 100644 --- a/x2paddle/op_mapper/paddle2onnx/opset9/opset.py +++ b/x2paddle/op_mapper/paddle2onnx/opset9/opset.py @@ -174,7 +174,8 @@ class OpSet9(object): inputs=[op.input('X')[0], temp_value], outputs=op.output('Out')) return [shape_node, y_node, node] - elif axis == -1 or axis == 0 or axis == (len(x_shape) - 1): + elif axis == -1 or axis == (len(x_shape) - 1 + ) or len(x_shape) == len(y_shape): node = helper.make_node( 'Add', inputs=[op.input('X')[0], op.input('Y')[0]], @@ -203,7 +204,8 @@ class OpSet9(object): inputs=[op.input('X')[0], temp_value], outputs=op.output('Out')) return [shape_node, y_node, node] - elif axis == -1 or axis == 0 or axis == (len(x_shape) - 1): + elif axis == -1 or axis == (len(x_shape) - 1 + ) or len(x_shape) == len(y_shape): node = helper.make_node( 'Sub', inputs=[op.input('X')[0], op.input('Y')[0]], @@ -565,7 +567,7 @@ class OpSet9(object): input_shape = block.vars[op.input('X')[0]].shape if op.attr('align_corners') or op.attr('align_mode') == 0: raise Exception( - "Resize in onnx(opset<=10) only support coordinate_transformation_mode: 'asymmetric', Try converting with --onnx_opest 11" + "Resize in onnx(opset<=10) only support coordinate_transformation_mode: 'asymmetric', Try converting with --onnx_opset 11" ) if ('OutSize' in input_names and len(op.input('OutSize')) > 0) or ( 'SizeTensor' in input_names and @@ -763,7 +765,8 @@ class OpSet9(object): inputs=[op.input('X')[0], temp_value], outputs=op.output('Out')) return [shape_node, y_node, node] - elif axis == -1 or axis == 0 or axis == (len(x_shape) - 1): + elif axis == -1 or axis == (len(x_shape) - 1 + ) or len(x_shape) == len(y_shape): node = helper.make_node( 'Mul', inputs=[op.input('X')[0], op.input('Y')[0]], -- GitLab