Skip to content

Commit f103eb5

Browse files
committed
fix summary when input node is not created
1 parent 0e8faa0 commit f103eb5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tf2onnx/graph.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -163,8 +163,9 @@ def summary(self):
163163
if self.input:
164164
lines.append("Inputs:")
165165
for name in self.input:
166-
lines.append("\t{}={}, {}, {}".format(name, g.get_node_by_output(name).type, g.get_shape(name),
167-
g.get_dtype(name)))
166+
node = g.get_node_by_output(name)
167+
op = node.type if node else "N/A"
168+
lines.append("\t{}={}, {}, {}".format(name, op, g.get_shape(name), g.get_dtype(name)))
168169

169170
if self.output:
170171
for name in self.output:
@@ -457,7 +458,7 @@ def make_node(self, op_type, inputs, attr=None, output_count=1, outputs=None, sk
457458
if op_name_scope:
458459
name = "_".join([op_name_scope, name])
459460

460-
logger.debug("Making node: %s", name)
461+
logger.debug("Making node: Name=%s, OP=%s", name, op_type)
461462

462463
if outputs is None:
463464
outputs = [name + ":" + str(i) for i in range(output_count)]

0 commit comments

Comments
 (0)