Skip to content

Commit 831d42e

Browse files
author
wayuanho
authored
Merge pull request #638 from zhijxu-MS/PR_branch
fix bug
2 parents 2a5e655 + b8c55ae commit 831d42e

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
@@ -199,7 +199,7 @@ def _handle_node_having_branches(self, node):
199199
# otherwise, it would impact their other output nodes
200200
if self._nodes_has_single_consumer_node(node.inputs):
201201
self._create_transpose_pairs_after_node(node)
202-
input_transposes = node.inputs
202+
input_transposes = set(node.inputs)
203203
for n in input_transposes:
204204
n_input = n.input[0]
205205
utils.make_sure(len(n.output) == 1, "only expect single output")

0 commit comments

Comments
 (0)