提交 31823494 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Forgot grad and R_op for allow_gc.

上级 d0bb8089
......@@ -1882,6 +1882,7 @@ class Scan(PureOp):
else:
info['name'] = None
info['mode'] = self.mode
info['allow_gc'] = self.allow_gc
outer_inputs = ([grad_steps] +
outer_inp_seqs +
......@@ -2047,6 +2048,7 @@ class Scan(PureOp):
else:
info['name'] = None
info['mode'] = self.mode
info['allow_gc'] = self.allow_gc
info['mit_mot_out_slices'] = self.mit_mot_out_slices * 2
info['destroy_map'] = OrderedDict()
new_tap_array = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论