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 03a2587 commit 9bb7013Copy full SHA for 9bb7013
onnx_diagnostic/helpers/dot_helper.py
@@ -157,8 +157,9 @@ def _mkn(obj: object) -> int:
157
edge_label[inp.name] = _make_edge_label(inp, multi_line=True)
158
159
# Small constant --> initializer
160
+ output_names = {n.name for n in outputs}
161
for node in nodes:
- if node.op_type != "Constant":
162
+ if node.op_type != "Constant" or node.output[0] in output_names:
163
continue
164
skip = False
165
for att in node.attribute:
0 commit comments