Searched refs:linalg_solve_is_vector_rhs (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 541 inline bool linalg_solve_is_vector_rhs(const Tensor& input, const Tensor& other) { in linalg_solve_is_vector_rhs() function
|
| H A D | BatchLinearAlgebra.cpp | 516 const bool vector_case = at::native::linalg_solve_is_vector_rhs(A, B); in TORCH_META_FUNC()
1900 const bool vector_case = at::native::linalg_solve_is_vector_rhs(LU, B); in TORCH_IMPL_FUNC()
3358 bool vector_case = linalg_solve_is_vector_rhs(input, other); in linalg_lstsq_out_info()
3544 bool vector_case = linalg_solve_is_vector_rhs(input, other); in linalg_lstsq_out()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 873 auto is_vector_case = at::native::linalg_solve_is_vector_rhs(dA, dB); in generic_solve_jvp()
3875 const bool vector_case = at::native::linalg_solve_is_vector_rhs(A, B_); in linalg_lstsq_backward()
5849 const bool vector_case = at::native::linalg_solve_is_vector_rhs(LU, X); in linalg_solve_jvp()
5888 const bool vector_case = at::native::linalg_solve_is_vector_rhs(LU, X); in linalg_solve_backward()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 1390 def linalg_solve_is_vector_rhs(input: Tensor, other: Tensor) -> bool: function
1418 vector_case = linalg_solve_is_vector_rhs(A, B)
|