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