提交 919d3ebc authored 作者: Olivier Breuleux's avatar Olivier Breuleux

fixed printing bug in gof.graph.as_string

上级 d77ccbce
...@@ -262,6 +262,8 @@ def as_string(i, o, ...@@ -262,6 +262,8 @@ def as_string(i, o,
exist for viewing convenience). exist for viewing convenience).
""" """
orph = orphans(i, o)
multi = set() multi = set()
seen = set() seen = set()
for output in o: for output in o:
...@@ -273,7 +275,7 @@ def as_string(i, o, ...@@ -273,7 +275,7 @@ def as_string(i, o,
for op in ops(i, o): for op in ops(i, o):
for input in op.inputs: for input in op.inputs:
op2 = input.owner op2 = input.owner
if input in i or op2 is None: if input in i or input in orph or op2 is None:
continue continue
if op2 in seen: if op2 in seen:
multi.add(op2) multi.add(op2)
...@@ -286,7 +288,7 @@ def as_string(i, o, ...@@ -286,7 +288,7 @@ def as_string(i, o,
return multi.index(x) + 1 return multi.index(x) + 1
def describe(r): def describe(r):
if r.owner is not None and r not in i: if r.owner is not None and r not in i and r not in orph:
op = r.owner op = r.owner
idx = op.outputs.index(r) idx = op.outputs.index(r)
if idx == op._default_output_idx: if idx == op._default_output_idx:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论