Skip to content

Commit 396b40b

Browse files
author
Joshua Monson
committed
fix failing tests
1 parent a6df660 commit 396b40b

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/qonnx/transformation/general.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,8 @@ def apply(self, model):
248248
# check if node inputs are connected to graph inputs or initializers
249249
# if so, we can keep the node in the graph
250250
for name in n.input:
251-
if model.get_initializer(name) or util.get_by_name(model.graph.input, name, "name"):
251+
if util.get_by_name(model.graph.initializer, name) or \
252+
util.get_by_name(model.graph.input, name):
252253
# this node is connected to graph inputs or initializers
253254
# so we can keep it in the graph
254255
graph_dependencies[node_idx] = set()

0 commit comments

Comments
 (0)