We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a6df660 commit 396b40bCopy full SHA for 396b40b
1 file changed
src/qonnx/transformation/general.py
@@ -248,7 +248,8 @@ def apply(self, model):
248
# check if node inputs are connected to graph inputs or initializers
249
# if so, we can keep the node in the graph
250
for name in n.input:
251
- if model.get_initializer(name) or util.get_by_name(model.graph.input, name, "name"):
+ if util.get_by_name(model.graph.initializer, name) or \
252
+ util.get_by_name(model.graph.input, name):
253
# this node is connected to graph inputs or initializers
254
# so we can keep it in the graph
255
graph_dependencies[node_idx] = set()
0 commit comments