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