Skip to content

Commit 7b244e0

Browse files
committed
remove unexisting nodes
1 parent d9ea76a commit 7b244e0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tf2onnx/graph.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1313,7 +1313,9 @@ def replace_all_inputs(self, ops, old_input, new_input):
13131313
if ops is not None:
13141314
keep_ops = True
13151315
elif old_input in self._input_to_node_name:
1316-
ops = [self.get_node_by_name(n) for n in self._input_to_node_name[old_input]]
1316+
ops = list(
1317+
filter(lambda a: a is not None,
1318+
map(self.get_node_by_name, self._input_to_node_name[old_input])))
13171319
keep_ops = False
13181320
else:
13191321
ops = []

0 commit comments

Comments
 (0)