Skip to content

Commit 4167ad1

Browse files
committed
fix failing unit test
1 parent 1de2a95 commit 4167ad1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -684,7 +684,7 @@ def reset_nodes(self, ops):
684684
for i in self.inputs:
685685
if i.name.startswith('Placeholder'):
686686
continue
687-
if i.name not in self._input_to_node_name:
687+
if i.name not in self._input_to_node_name and i.name + ':0' not in self._input_to_node_name:
688688
raise ValueError("graph input %r not exist in graph." % i.name)
689689

690690
self._dtypes = remained_dtypes

0 commit comments

Comments
 (0)