def _axpbz(out, afac, a, other, queue=None): """Compute ``out = afac * a + other``, where `other` is a scalar.""" return elementwise.get_axpbz_kernel(out.context, out.dtype)
def _axpbz(out, a, x, b, queue=None): """Compute ``z = a * x + b``, where `b` is a scalar.""" a = np.array(a) b = np.array(b) return elementwise.get_axpbz_kernel(out.context, a.dtype, x.dtype, b.dtype, out.dtype)