提交 75246107 authored 作者: Roy Xue's avatar Roy Xue

Add assert string

上级 15198850
...@@ -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 assert ins is None, "Here we only support the possibility to view one input"
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]]
...@@ -848,7 +848,7 @@ class ProfileStats(object): ...@@ -848,7 +848,7 @@ class ProfileStats(object):
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, "Here we only support the possibility to view one input"
vidx = vmap[idx] vidx = vmap[idx]
assert len(vidx) == 1 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 ins is not None: if ins is not None:
# This is needed for destroy_map in case it # This is needed for destroy_map in case it
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论