Python get_numba_ufunc Exemples

Exemple #1

Fichier : blaze.py Projet : kastnerkyle/dask

 def compute_broadcast(expr, *data, **kwargs):
     leaves = expr._inputs
     expr_inds = tuple(range(ndim(expr)))[::-1]
     func = get_numba_ufunc(expr)
     return atop(func,
                 next(names), expr_inds,
                 *concat((dat, tuple(range(ndim(dat))[::-1])) for dat in data))

Exemple #2

Fichier : dask.py Projet : snmz216/blaze

 def compute_broadcast(expr, *data, **kwargs):
     expr_inds = tuple(range(ndim(expr)))[::-1]
     func = get_numba_ufunc(expr)
     return atop(func,
                 expr_inds,
                 *concat(
                     (dat, tuple(range(ndim(dat))[::-1])) for dat in data),
                 dtype=data[-1].dtype)