提交 fba9e145 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3595 from lamblin/fix_transfer_rop

Fix return value of R_op in HostFromGpu/GpuFromHost
...@@ -92,7 +92,7 @@ class HostFromGpu(GpuOp): ...@@ -92,7 +92,7 @@ class HostFromGpu(GpuOp):
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
ev, = eval_points ev, = eval_points
return self(ev) return [self(ev)]
def infer_shape(self, node, xshp): def infer_shape(self, node, xshp):
return xshp return xshp
...@@ -152,7 +152,7 @@ class GpuFromHost(GpuOp): ...@@ -152,7 +152,7 @@ class GpuFromHost(GpuOp):
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
ev, = eval_points ev, = eval_points
self(ev) [self(ev)]
def infer_shape(self, node, xshp): def infer_shape(self, node, xshp):
return xshp return xshp
......
...@@ -345,7 +345,7 @@ class HostFromGpu(Op): ...@@ -345,7 +345,7 @@ class HostFromGpu(Op):
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
ev, = eval_points ev, = eval_points
return self(ev) return [self(ev)]
def infer_shape(self, node, xshp): def infer_shape(self, node, xshp):
return xshp return xshp
...@@ -386,7 +386,7 @@ class GpuFromHost(Op): ...@@ -386,7 +386,7 @@ class GpuFromHost(Op):
def R_op(self, inputs, eval_points): def R_op(self, inputs, eval_points):
ev, = eval_points ev, = eval_points
return self(ev) return [self(ev)]
def infer_shape(self, node, xshp): def infer_shape(self, node, xshp):
return xshp return xshp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论