Skip to content

Commit 9bb7013

Browse files
authored
minor fix (#335)
* improves dot * improves dot rendering * fix last bug
1 parent 03a2587 commit 9bb7013

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onnx_diagnostic/helpers/dot_helper.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,8 +157,9 @@ def _mkn(obj: object) -> int:
157157
edge_label[inp.name] = _make_edge_label(inp, multi_line=True)
158158

159159
# Small constant --> initializer
160+
output_names = {n.name for n in outputs}
160161
for node in nodes:
161-
if node.op_type != "Constant":
162+
if node.op_type != "Constant" or node.output[0] in output_names:
162163
continue
163164
skip = False
164165
for att in node.attribute:

0 commit comments

Comments
 (0)