From c551408f76e9f1b40c64b4d5912b5d7a0643e62b Mon Sep 17 00:00:00 2001 From: TomWildenhain-Microsoft <67606533+TomWildenhain-Microsoft@users.noreply.github.com> Date: Wed, 10 Feb 2021 23:20:54 -0500 Subject: [PATCH] Make merge optimizer faster (#1326) Signed-off-by: Tom Wildenhain --- tf2onnx/optimizer/merge_duplicated_nodes_optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf2onnx/optimizer/merge_duplicated_nodes_optimizer.py b/tf2onnx/optimizer/merge_duplicated_nodes_optimizer.py index cadd4c9d4..8d876ce0c 100644 --- a/tf2onnx/optimizer/merge_duplicated_nodes_optimizer.py +++ b/tf2onnx/optimizer/merge_duplicated_nodes_optimizer.py @@ -102,7 +102,7 @@ def _merge_nodes_that_are_duplicated(self, nodes_to_process, graph): if set(node_to_delete.output).intersection(set(graph.outputs)): continue for old_input, new_input in zip(node_to_delete.output, node_to_retain.output): - graph.replace_all_inputs(old_input, new_input, ops=graph.get_nodes()) + graph.replace_all_inputs(old_input, new_input) graph.remove_node(node_to_delete.name) self._graph_can_be_optimized = True