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