提交 aab78e7e authored 作者: sentient07's avatar sentient07

Changed node.inputs to fgraph.variables

上级 c1c990cd
...@@ -1304,7 +1304,7 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1304,7 +1304,7 @@ class LocalOptGroup(LocalOptimizer):
if not repl: if not repl:
continue continue
else: else:
self.node_created[opt] += len(graph.ops(node.inputs, repl)) self.node_created[opt] += len(graph.ops(fgraph.variables, repl))
self.applied_true[opt] += 1 self.applied_true[opt] += 1
if not multiple_opts or not repl[0].owner: if not multiple_opts or not repl[0].owner:
return repl return repl
...@@ -1312,8 +1312,6 @@ class LocalOptGroup(LocalOptimizer): ...@@ -1312,8 +1312,6 @@ class LocalOptGroup(LocalOptimizer):
# Ensuring not the input of graph # Ensuring not the input of graph
assert repl[0].owner assert repl[0].owner
new_node = repl[0].owner new_node = repl[0].owner
if hasattr(new_node, 'fgraph'):
apply_mult_opts(new_node, new_node.fgraph, True)
apply_mult_opts(new_node, fgraph, True) apply_mult_opts(new_node, fgraph, True)
return repl return repl
node_start = time.time() node_start = time.time()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论