提交 090b9a0f authored 作者: Pascal Lamblin's avatar Pascal Lamblin

[mq]: chose

上级 fe5a9a35
......@@ -286,7 +286,7 @@ def local_gpu_rebroadcast(node):
'''rebroadcast(host_from_gpu(x)) -> host_from_gpu(rebroadcast(x))'''
if isinstance(node.op, tensor.Rebroadcast):
x, = node.inputs
if (x.owner and x.owner.op == host_fom_gpu):
if (x.owner and x.owner.op == host_from_gpu):
gpu_x = x.owner.inputs[0]
return [host_from_gpu(node.op(gpu_x))]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论