We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1de2a95 commit 4167ad1Copy full SHA for 4167ad1
tf2onnx/graph.py
@@ -684,7 +684,7 @@ def reset_nodes(self, ops):
684
for i in self.inputs:
685
if i.name.startswith('Placeholder'):
686
continue
687
- if i.name not in self._input_to_node_name:
+ if i.name not in self._input_to_node_name and i.name + ':0' not in self._input_to_node_name:
688
raise ValueError("graph input %r not exist in graph." % i.name)
689
690
self._dtypes = remained_dtypes
0 commit comments