提交 d6abf92c authored 作者: Frederic Bastien's avatar Frederic Bastien

One more step when triing to reuse the output of GpuConvOp.

上级 223e351f
...@@ -184,6 +184,7 @@ class GpuConv(Op): ...@@ -184,6 +184,7 @@ class GpuConv(Op):
print "out", out print "out", out
out[0] = cuda_ndarray.conv(img, kern, out[0] = cuda_ndarray.conv(img, kern,
mode=self.border_mode, mode=self.border_mode,
out=out[0],
subsample=self.subsample, subsample=self.subsample,
logical_img_shape=self.logical_img_hw, logical_img_shape=self.logical_img_hw,
logical_kern_shape=self.logical_kern_hw, logical_kern_shape=self.logical_kern_hw,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论