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