Added Identity node to the removal list

fastmachinelearning · Sep 12, 2024 · 8bad7e7 · 8bad7e7
1 parent 93acc3a
commit 8bad7e7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/qonnx/transformation/remove.py b/src/qonnx/transformation/remove.py
@@ -138,5 +138,9 @@ def apply(self, model):
                     remove_node_and_rewire(model, n)
                     graph_modified = True
                     break
+            elif n.op_type == "Identity":
+                remove_node_and_rewire(model, n)
+                graph_modified = True
+                break
         model = model.transform(InferShapes())
         return (model, graph_modified)