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 D | BatchLinearAlgebra.cpp | 2460 …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 D | linalg.h | 378 return torch::linalg_solve_triangular_out( in solve_triangular_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | BatchLinearAlgebra.cpp | 3813 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 D | generated_ops.cpp | 2835 at::native::linalg_solve_triangular_out( in __anon5d9c3eb910e02()
|