From cad8807d23c8fa053600a01b5728f83c4c052c72 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 6 Jul 2020 12:30:34 +0100
Subject: [PATCH] [Transform] introduce RemoveStaticGraphInputs
---
src/finn/transformation/general.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/src/finn/transformation/general.py b/src/finn/transformation/general.py
index 9a238ecf1..d9147b700 100644
--- a/src/finn/transformation/general.py
+++ b/src/finn/transformation/general.py
@@ -57,6 +57,21 @@ class RemoveUnusedInitAndValueInfo(Transformation):
return (model, graph_modified)
+class RemoveStaticGraphInputs(Transformation):
+ "Remove any top-level graph inputs that have initializers."
+
+ def apply(self, model):
+ graph_modified = False
+ for i in model.graph.input:
+ if model.get_initializer(i.name) is not None:
+ # move ValueInfo to internal (value_info) container
+ model.graph.value_info.append(i)
+ model.graph.input.remove(i)
+ graph_modified = True
+
+ return (model, graph_modified)
+
+
class GiveUniqueNodeNames(Transformation):
"""Give unique names to each node in the graph using enumeration."""
--
GitLab