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)