提交 a11b232f authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Remove the 0-ndim restriction on GpuElemwise.

上级 133b8a48
......@@ -208,8 +208,6 @@ def local_gpuflatten(node):
def local_gpu_elemwise(node):
op = node.op
name = op.name
if node.outputs[0].ndim == 0:
return
if name:
name = 'Gpu'+name
res = GpuElemwise(op.scalar_op, name=name,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论