

def _schedule(self, inputs, out):
    input_shapes = [a.shape for a in inputs]
    schedule = self.engine.schedule(input_shapes)
    idtypes = tuple((i.dtype for i in inputs))
    try:
        (outdtype, kernel) = self.kernelmap[idtypes]
    except KeyError:
        idtypes = self._search_matching_signature(idtypes)
        (outdtype, kernel) = self.kernelmap[idtypes]
    if ((out is not None) and (schedule.output_shapes[0] != out.shape)):
        raise ValueError('output shape mismatch')
    return (idtypes, schedule, outdtype, kernel)

dataset/ETHPy150Open numba/numba/numba/npyufunc/deviceufunc.py GenerializedUFunc._schedule/original