Skip to content

Commit 8bad7e7

Browse files
jvrecamaltanar
authored andcommitted
Added Identity node to the removal list
1 parent 93acc3a commit 8bad7e7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/qonnx/transformation/remove.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,5 +138,9 @@ def apply(self, model):
138138
remove_node_and_rewire(model, n)
139139
graph_modified = True
140140
break
141+
elif n.op_type == "Identity":
142+
remove_node_and_rewire(model, n)
143+
graph_modified = True
144+
break
141145
model = model.transform(InferShapes())
142146
return (model, graph_modified)

0 commit comments

Comments
 (0)