提交 a99ef2a1 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix bug reported by abalkin

上级 b412823a
...@@ -837,9 +837,9 @@ class ScanSaveMem(gof.Optimizer): ...@@ -837,9 +837,9 @@ class ScanSaveMem(gof.Optimizer):
if not (idx in replaced_outs) and not idx in not_required: if not (idx in replaced_outs) and not idx in not_required:
nw_pos = compress_map[idx] nw_pos = compress_map[idx]
old_new += [(o, new_outs[nw_pos])] old_new += [(o, new_outs[nw_pos])]
remove = [ x[0].owner for x in old_new]
fgraph.replace_all_validate_remove(old_new, fgraph.replace_all_validate_remove(old_new,
remove=[node], remove=remove,
reason='scan_save_mem') reason='scan_save_mem')
def apply(self, fgraph): def apply(self, fgraph):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论