提交 2d8742b4 authored 作者: --global's avatar --global

Set borrow=True for inputs when not preallocating outputs

上级 b3f1452f
...@@ -723,9 +723,15 @@ class Scan(PureOp): ...@@ -723,9 +723,15 @@ class Scan(PureOp):
self.n_mit_sot + self.n_mit_sot +
self.n_sit_sot + self.n_sit_sot +
self.n_nit_sot) self.n_nit_sot)
wrapped_inputs = [Param(x, borrow=False) for x in self.inputs] if theano.config.scan.allow_output_prealloc:
borrow_outputs = theano.config.scan.allow_output_prealloc wrapped_inputs = [Param(x, borrow=False) for x in
wrapped_outputs = [Out(x, borrow=borrow_outputs) for x in self.inputs]
wrapped_outputs = [Out(x, borrow=True) for x in
self.outputs[:slices]]
else:
wrapped_inputs = [Param(x, borrow=True) for x in
self.inputs]
wrapped_outputs = [Out(x, borrow=False) for x in
self.outputs[:slices]] self.outputs[:slices]]
wrapped_outputs += self.outputs[slices:] wrapped_outputs += self.outputs[slices:]
profile = None profile = None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论