提交 023876a0 authored 作者: Roy Xue's avatar Roy Xue

Code clean up

上级 405397d0
...@@ -713,7 +713,7 @@ class ProfileStats(object): ...@@ -713,7 +713,7 @@ class ProfileStats(object):
assert len(vidx) == 1, "Here we only support the possibility to destroy one input" assert len(vidx) == 1, "Here we only support the possibility to destroy one input"
ins = node.inputs[vidx[0]] ins = node.inputs[vidx[0]]
if vmap and idx2 in vmap: if vmap and idx2 in vmap:
assert ins is None, "Here we only support the possibility to view one input" assert ins is None
vidx = vmap[idx2] vidx = vmap[idx2]
assert len(vidx) == 1, "Here we only support the possibility to view one input" assert len(vidx) == 1, "Here we only support the possibility to view one input"
ins = node.inputs[vidx[0]] ins = node.inputs[vidx[0]]
...@@ -843,7 +843,7 @@ class ProfileStats(object): ...@@ -843,7 +843,7 @@ class ProfileStats(object):
assert len(vidx) == 1, "Here we only support the possibility to destroy one input" assert len(vidx) == 1, "Here we only support the possibility to destroy one input"
ins = node.inputs[vidx[0]] ins = node.inputs[vidx[0]]
if vmap and idx in vmap: if vmap and idx in vmap:
assert ins is None, "Here we only support the possibility to view one input" assert ins is None
vidx = vmap[idx] vidx = vmap[idx]
assert len(vidx) == 1, "Here we only support the possibility to destroy one input" assert len(vidx) == 1, "Here we only support the possibility to destroy one input"
ins = node.inputs[vidx[0]] ins = node.inputs[vidx[0]]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论