提交 48d21c36 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix scan optimization

上级 588dcf8f
...@@ -1159,7 +1159,7 @@ class ScanMerge(gof.Optimizer): ...@@ -1159,7 +1159,7 @@ class ScanMerge(gof.Optimizer):
Questionable, we should also consider profile ? Questionable, we should also consider profile ?
""" """
rep = set_nodes[0] rep = set_nodes[0]
if not rep.op.as_while and node.op.as_while: if rep.op.as_while != node.op.as_while:
return False return False
nsteps = node.inputs[0] nsteps = node.inputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论