From a59e5794588a59fa78500859fddba13e12fb0dae Mon Sep 17 00:00:00 2001 From: Wei Chu Date: Fri, 15 Jan 2021 11:40:22 -0800 Subject: [PATCH] fix sanity --- .../mxnet/contrib/onnx/mx2onnx/_op_translations.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py b/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py index 2027c6bf3934..0a633d879f80 100644 --- a/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py +++ b/python/mxnet/contrib/onnx/mx2onnx/_op_translations.py @@ -3014,7 +3014,7 @@ def convert_reshape_like(node, **kwargs): if lhs_end >= 0: nodes += [ create_tensor([lhs_end], name+'_lhs_end', kwargs["initializer"]), - ] + ] else: nodes += [ create_tensor([lhs_end], name+'_lhs_end_neg', kwargs["initializer"]), @@ -3030,7 +3030,7 @@ def convert_reshape_like(node, **kwargs): if rhs_end >= 0: nodes += [ create_tensor([rhs_end], name+'_rhs_end', kwargs["initializer"]), - ] + ] else: nodes += [ create_tensor([rhs_end], name+'_rhs_end_neg', kwargs["initializer"]), @@ -3038,16 +3038,12 @@ def convert_reshape_like(node, **kwargs): ] nodes += [ - make_node('Slice', [name+'_lhs_shape', name+'_0', name+'_lhs_begin'], - [name+'_slice0_out']), - make_node('Slice', [name+'_rhs_shape', name+'_rhs_begin', name+'_rhs_end'], - [name+'_slice1_out']), + make_node('Slice', [name+'_lhs_shape', name+'_0', name+'_lhs_begin'], [name+'_slice0_out']), + make_node('Slice', [name+'_rhs_shape', name+'_rhs_begin', name+'_rhs_end'], [name+'_slice1_out']), make_node('Concat', [name+'_slice0_out', name+'_slice1_out'], [name+'_concat0_out'], axis=0), - make_node('Slice', [name+'_lhs_shape', name+'_lhs_end', name+'_lhs_dim'], - [name+'_slice2_out']), + make_node('Slice', [name+'_lhs_shape', name+'_lhs_end', name+'_lhs_dim'], [name+'_slice2_out']), make_node('Concat', [name+'_concat0_out', name+'_slice2_out'], [name+'_concat1_out'], axis=0), make_node('Reshape', [lhs, name+'_concat1_out'], [name], name=name) ] return nodes -