for n in graph.nodes():
attrs = {k: n[k] for k in n.attributeNames()}
attrs = str(attrs).replace(""", " ") // singlequote will be escaped by tensorboard
if any(i.uniqueName() not in scope.keys() for i in n.inputs()): // 0.3.1 workaround
continue
inputs = [replace(i.uniqueName(), scope) for i in n.inputs()]
outputnode = next(iter(n.outputs())) // FIXME: only first output is considered
uname = outputnode.uniqueName()
if outputnode.type().kind() == "TensorType":
After Change
"attr": str(n.type()),
"outputsize": outputsize})
mapping = {}
for n in nodes:
mapping[n["name"]] = scope[n["name"]] + "/" + \
n["op"].replace("onnx::", "") + "_" + n["name"]
for n in nodes: