提交 784bab43 authored 作者: Nicolas Ballas's avatar Nicolas Ballas

handle gpu input/outpus

上级 46d60030
...@@ -507,6 +507,8 @@ class AbstractConv2d(BaseAbstractConv2d): ...@@ -507,6 +507,8 @@ class AbstractConv2d(BaseAbstractConv2d):
def perform(self, node, inp, out_): def perform(self, node, inp, out_):
img, kern = inp img, kern = inp
img = numpy.asarray(img)
kern = numpy.asarray(kern)
o, = out_ o, = out_
mode = self.border_mode mode = self.border_mode
...@@ -527,7 +529,7 @@ class AbstractConv2d(BaseAbstractConv2d): ...@@ -527,7 +529,7 @@ class AbstractConv2d(BaseAbstractConv2d):
conv_out = self.corr2d(img, kern, mode) conv_out = self.corr2d(img, kern, mode)
conv_out =conv_out[:, :, ::self.subsample[0], ::self.subsample[1]] conv_out =conv_out[:, :, ::self.subsample[0], ::self.subsample[1]]
o[0] = conv_out o[0] = node.outputs[0].type.filter(conv_out)
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论