def lu_impl(operand):
lu, pivot = xla.apply_primitive(lu_p, operand)
return core.pack((lu, pivot))
def qr_impl(operand, full_matrices):
q, r = xla.apply_primitive(qr_p, operand, full_matrices=full_matrices)
return core.pack((q, r))
def svd_impl(operand, full_matrices, compute_uv):
s, u, vt = xla.apply_primitive(svd_p, operand, full_matrices=full_matrices,
compute_uv=compute_uv)
return s, u, vt
def eigh_impl(operand, lower):
v, w = xla.apply_primitive(eigh_p, operand, lower=lower)
return core.pack((v, w))
def _lu_impl(operand): lu, pivot = xla.apply_primitive(lu_p, operand) return lu, pivot
def eigh_impl(operand, lower): v, w = xla.apply_primitive(eigh_p, operand, lower=lower) return v, w
def eig_impl(operand): return xla.apply_primitive(eig_p, operand)
def fft_impl(x, fft_type, fft_lengths):
return xla.apply_primitive(fft_p,
x,
fft_type=fft_type,
fft_lengths=fft_lengths)
def eig_impl(operand, *, compute_left_eigenvectors, compute_right_eigenvectors):
return (
xla.apply_primitive(eig_p, operand,
compute_left_eigenvectors=compute_left_eigenvectors,
compute_right_eigenvectors=compute_right_eigenvectors))
