提交 9f058b8b authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix return value of R_op in HostFromGpu/GpuFromHost

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