提交 532dfb3a authored 作者: Frederic Bastien's avatar Frederic Bastien

Partially undo dde50054. We should merge parallel scan what ever their number of inputs and outputs

上级 37172bad
...@@ -1819,8 +1819,6 @@ class ScanMerge(gof.Optimizer): ...@@ -1819,8 +1819,6 @@ class ScanMerge(gof.Optimizer):
""" """
rep = set_nodes[0] rep = set_nodes[0]
if (rep.op.as_while != node.op.as_while or if (rep.op.as_while != node.op.as_while or
len(rep.inputs) != len(node.inputs) or
len(rep.outputs) != len(node.outputs) or
node.op.truncate_gradient != rep.op.truncate_gradient or node.op.truncate_gradient != rep.op.truncate_gradient or
node.op.mode != rep.op.mode): node.op.mode != rep.op.mode):
return False return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论