提交 c27b80ed authored 作者: nouiz's avatar nouiz

Merge pull request #868 from pascanur/fix_pushout_opt

Change replace_all_validate to replace_all_validate_remove
......@@ -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 replace_with:
fgraph.replace_all_validate_remove(
replace_with.items(),
remove=[node],
reason='scan_push_computation_out')
else:
return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论