Skip to content

Commit 7047b47

Browse files
committed
Update transpose_optimizer.py
1 parent 4949509 commit 7047b47

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ def _handle_nhwc_tranpose(self, trans):
283283
# move transpose into branches to let Transposes can be "handled" in each branch
284284
for n in out_nodes:
285285
branch_trans = n.graph.make_node("Transpose", [trans.input[0]], attr=trans.attr_onnx)
286-
n.graph.replace_input(n, trans.output[0], branch_trans.output[0], 0)
286+
n.graph.replace_input(n, trans.output[0], branch_trans.output[0])
287287

288288
self._g.remove_node(trans.name)
289289
return False

0 commit comments

Comments
 (0)