Skip to content

Commit c4bce3a

Browse files
committed
fix an issue when subgraph output node has more than one output
1 parent d2baed0 commit c4bce3a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def __init__(self, nodes, output_shapes=None, dtypes=None, target=None, opset=No
369369
body_graphs = n.graph.contained_graphs.pop(n.name, None)
370370
self.remove_node(n.name)
371371

372-
new_outputs = [o if o != output else new_output_name for output in n.output]
372+
new_outputs = [output if output != o else new_output_name for output in n.output]
373373
new_node = self.make_node(n.type, n.input, outputs=new_outputs, attr=n.attr, name=n.name,
374374
skip_conversion=n._skip_conversion, dtypes=n_dtypes, shapes=n_shapes)
375375

0 commit comments

Comments
 (0)