Skip unregistering optional inputs/outputs that are empty (onnx#1467)

Signed-off-by: Tom Wildenhain <tomwi@microsoft.com>
zerollzeng · May 16, 2021 · 9c2f25f · 9c2f25f
1 parent 3b1a0d8
commit 9c2f25f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tf2onnx/graph.py b/tf2onnx/graph.py
@@ -691,6 +691,8 @@ def remove_node(self, node_name):
             self.inputs.remove(node)
 
         for op_output in node.output:
+            if op_output == "":
+                continue
             del self._output_to_node_name[op_output]
 
             if op_output in self._output_shapes:
@@ -699,6 +701,8 @@ def remove_node(self, node_name):
                 del self._dtypes[op_output]
 
         for op_input in node.input:
+            if op_input == "":
+                continue
             utils.make_sure(
                 op_input in self._output_to_consumers,
                 "Input %r of node %r not found.", op_input, node_name)