Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit ce777ea

Browse files
author
Wei Chu
committed
fix sanity
1 parent 161a748 commit ce777ea

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

python/mxnet/contrib/onnx/mx2onnx/_op_translations.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3118,7 +3118,7 @@ def convert_reshape_like(node, **kwargs):
31183118
if lhs_end >= 0:
31193119
nodes += [
31203120
create_tensor([lhs_end], name+'_lhs_end', kwargs["initializer"]),
3121-
]
3121+
]
31223122
else:
31233123
nodes += [
31243124
create_tensor([lhs_end], name+'_lhs_end_neg', kwargs["initializer"]),
@@ -3134,24 +3134,20 @@ def convert_reshape_like(node, **kwargs):
31343134
if rhs_end >= 0:
31353135
nodes += [
31363136
create_tensor([rhs_end], name+'_rhs_end', kwargs["initializer"]),
3137-
]
3137+
]
31383138
else:
31393139
nodes += [
31403140
create_tensor([rhs_end], name+'_rhs_end_neg', kwargs["initializer"]),
31413141
make_node('Add', [name+'_rhs_dim', name+'_rhs_end_neg'], [name+'_rhs_end']),
31423142
]
31433143

31443144
nodes += [
3145-
make_node('Slice', [name+'_lhs_shape', name+'_0', name+'_lhs_begin'],
3146-
[name+'_slice0_out']),
3147-
make_node('Slice', [name+'_rhs_shape', name+'_rhs_begin', name+'_rhs_end'],
3148-
[name+'_slice1_out']),
3145+
make_node('Slice', [name+'_lhs_shape', name+'_0', name+'_lhs_begin'], [name+'_slice0_out']),
3146+
make_node('Slice', [name+'_rhs_shape', name+'_rhs_begin', name+'_rhs_end'], [name+'_slice1_out']),
31493147
make_node('Concat', [name+'_slice0_out', name+'_slice1_out'], [name+'_concat0_out'], axis=0),
3150-
make_node('Slice', [name+'_lhs_shape', name+'_lhs_end', name+'_lhs_dim'],
3151-
[name+'_slice2_out']),
3148+
make_node('Slice', [name+'_lhs_shape', name+'_lhs_end', name+'_lhs_dim'], [name+'_slice2_out']),
31523149
make_node('Concat', [name+'_concat0_out', name+'_slice2_out'], [name+'_concat1_out'], axis=0),
31533150
make_node('Reshape', [lhs, name+'_concat1_out'], [name], name=name)
31543151
]
31553152

31563153
return nodes
3157-

0 commit comments

Comments
 (0)