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))
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)