提交 25ca446b authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix syntax and flake8 errors in opt_util.py

上级 17f74589
......@@ -31,8 +31,8 @@ def grab_cpu_scalar(v, nd):
"""
if v.owner is not None:
n = v.owner
elif (isinstance(n.op, (GpuDimShuffle, DimShuffle)) and
n.op.new_order == ('x',) * nd):
if (isinstance(n.op, (GpuDimShuffle, DimShuffle)) and
n.op.new_order == ('x',) * nd):
return grab_cpu_scalar(n.inputs[0], n.inputs[0].ndim)
elif isinstance(n.op, (GpuFromHost, HostFromGpu)):
return grab_cpu_scalar(n.inputs[0], nd)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论