提交 2f4daa6b authored 作者: Roy Xue's avatar Roy Xue

check_node_state fix

上级 fdf9bd39
......@@ -783,11 +783,12 @@ class ProfileStats(object):
"""
inputs = node.inputs
outputs = node.outputs
deps = inputs + node.destroy_dependencies
# TODO: Move at compute_map creation to speed things up.
for node in inputs:
if isinstance(node, graph.Constant):
compute_map[node][0] = 1
computed_ins = all(compute_map[v][0] for v in inputs)
computed_ins = all(compute_map[v][0] for v in deps)
if computed_ins:
return True
else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论