def make_node(self, x, axis, splits):
node = Split.make_node(self, x, axis, splits)
x = as_gpuarray_variable(x, infer_context_name(x))
outs = [GpuArrayType(dtype=o.dtype, broadcastable=o.broadcastable,
context_name=x.type.context_name)()
for o in node.outputs]
return Apply(self, [x] + node.inputs[1:], outs)
def make_node(self, x, axis, splits):
node = Split.make_node(self, x, axis, splits)
x = as_gpuarray_variable(x, infer_context_name(x))
outs = [GpuArrayType(dtype=o.dtype, broadcastable=o.broadcastable,
context_name=x.type.context_name)()
for o in node.outputs]
return Apply(self, [x] + node.inputs[1:], outs)