Home
last modified time | relevance | path

Searched refs:linalg_solve_triangular_out (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DBatchLinearAlgebra.cpp2460 …at::linalg_solve_triangular_out(const_cast<Tensor&>(B), *LU_, B1, /*upper=*/false, /*left=*/true, … in lu_solve_kernel()
2462 at::linalg_solve_triangular_out(const_cast<Tensor&>(B), *LU_, B, /*upper=*/true); in lu_solve_kernel()
2466 at::linalg_solve_triangular_out(const_cast<Tensor&>(B), LU_H, B, /*upper=*/false); in lu_solve_kernel()
2468 …at::linalg_solve_triangular_out(const_cast<Tensor&>(B), LU_H, B, /*upper=*/true, /*left=*/true, /*… in lu_solve_kernel()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dlinalg.h378 return torch::linalg_solve_triangular_out( in solve_triangular_out()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBatchLinearAlgebra.cpp3813 Tensor& linalg_solve_triangular_out( in linalg_solve_triangular_out() function
4007 linalg_solve_triangular_out(A, B, upper, left, unitriangular, out); in linalg_solve_triangular()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dgenerated_ops.cpp2835 at::native::linalg_solve_triangular_out( in __anon5d9c3eb910e02()