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

check_node_state fix

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