提交 0d2f82a5 authored 作者: Ben Poole's avatar Ben Poole

add missing return to GpuFromHost R_op

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