Searched refs:checkInputsSolver (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 137 inline void checkInputsSolver(const Tensor& A, in checkInputsSolver() function
|
| H A D | BatchLinearAlgebra.cpp | 520 at::native::checkInputsSolver(A, B_, /*left=*/left, "linalg.solve"); in TORCH_META_FUNC()
594 at::native::checkInputsSolver(LU, B, left, "linalg.lu_solve"); in TORCH_META_FUNC()
3820 checkInputsSolver(A, B, left, "linalg.solve_triangular"); in linalg_solve_triangular_out()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 809 def checkInputsSolver(A: Tensor, B: Tensor, left: bool, f_name: str): function
1173 checkInputsSolver(LU, B, left, "linalg.lu_solve")
1420 checkInputsSolver(A, B_, left, "linalg.solve")
1471 checkInputsSolver(A, B, left, "linalg.solve_triangular")
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | LinearAlgebra.mm | 716 checkInputsSolver(A, B, left, "linalg.solve_triangular");
|