Skip to content

Commit 0a87482

Browse files
committed
transpose optimizer should optime graphs in nodes' attr
1 parent a709f94 commit 0a87482

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,9 @@ def merge_duplicated_transposes(self):
131131
graph.delete_unused_nodes(graph.outputs)
132132

133133
def _optimize(self, graph):
134+
return self._apply_optimization(graph, self._optimize_at_current_graph_level)
135+
136+
def _optimize_at_current_graph_level(self, graph):
134137
self._g = graph
135138
self.pre_optimize_action()
136139
no_action = False

0 commit comments

Comments
 (0)