Update transpose_optimizer.py

onnx · Aug 4, 2020 · 7047b47 · 7047b47
1 parent 4949509
commit 7047b47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tf2onnx/optimizer/transpose_optimizer.py b/tf2onnx/optimizer/transpose_optimizer.py
@@ -283,7 +283,7 @@ def _handle_nhwc_tranpose(self, trans):
         # move transpose into branches to let Transposes can be "handled" in each branch
         for n in out_nodes:
             branch_trans = n.graph.make_node("Transpose", [trans.input[0]], attr=trans.attr_onnx)
-            n.graph.replace_input(n, trans.output[0], branch_trans.output[0], 0)
+            n.graph.replace_input(n, trans.output[0], branch_trans.output[0])
 
         self._g.remove_node(trans.name)
         return False