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

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

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