Skip to content

Commit c551408

Browse files
Make merge optimizer faster (#1326)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 5cfb2c9 commit c551408

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/optimizer/merge_duplicated_nodes_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def _merge_nodes_that_are_duplicated(self, nodes_to_process, graph):
102102
if set(node_to_delete.output).intersection(set(graph.outputs)):
103103
continue
104104
for old_input, new_input in zip(node_to_delete.output, node_to_retain.output):
105-
graph.replace_all_inputs(old_input, new_input, ops=graph.get_nodes())
105+
graph.replace_all_inputs(old_input, new_input)
106106
graph.remove_node(node_to_delete.name)
107107
self._graph_can_be_optimized = True
108108

0 commit comments

Comments
 (0)