5619f0a742cfee882b51b6798215362e227c2732,tensorboardX/graph.py,,parse,#Any#,14

Before Change

    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

After Change

    mapping = {}
    for n in nodes:
        mapping[n["name"]] = scope[n["name"]] + "/" + \
            n["op"].replace("onnx::", "") + "_" + n["name"]
    for n in nodes:
        n["name"] = mapping[n["name"]]
        for i, s in enumerate(n["inputs"]):
            n["inputs"][i] = mapping[s]

Instances