Skip to content

Commit f214a07

Browse files
committed
refactor
1 parent 30fa0a8 commit f214a07

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -426,12 +426,8 @@ def _identity_handler(self, trans, node):
426426

427427
def _concat_handler(self, trans, node):
428428
if self._handle_node_having_branches(node):
429-
perm = trans.get_attr("perm").ints
430-
axis_attr = node.get_attr("axis")
431-
if axis_attr:
432-
axis = axis_attr.i # fix
433-
else:
434-
axis = 0
429+
perm = trans.get_attr_value("perm")
430+
axis = node.get_attr_value("axis", 0)
435431
new_axis = perm[axis]
436432
node.set_attr("axis", new_axis)
437433
return True

0 commit comments

Comments
 (0)