Skip to content

Commit 0bb2540

Browse files
Fix bug in transpose optimizer where node is deleted twice (#1496)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 94ad674 commit 0bb2540

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,8 @@ def _optimize_at_current_graph_level(self, graph):
176176
# referencing already deleted elements
177177
break
178178

179-
if is_useless_transpose(n):
179+
# Make sure node wasn't already deleted in _handle_nhwc_tranpose
180+
if graph.get_node_by_name(n.name) is not None and is_useless_transpose(n):
180181
no_action = False
181182
iteration_cnt += 1
182183
self._remove_useless_tranpose(n)

0 commit comments

Comments
 (0)