提交 634d27c9 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

Change replace_all_validate to replace_all_validate_remove and fix the resulting error

上级 202eed7f
......@@ -297,8 +297,11 @@ class PushOutNonSeqScan(gof.Optimizer):
*shape)
# We need to add one extra dimension to the outputs
fgraph.replace_all_validate(replace_with.items(),
reason='scan_push_computation_out')
if len(replace_with.items()) > 0:
fgraph.replace_all_validate_remove(
replace_with.items(),
remove=[node],
reason='scan_push_computation_out')
else:
return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论