Home
last modified time | relevance | path

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 DLinearAlgebraUtils.h541 inline bool linalg_solve_is_vector_rhs(const Tensor& input, const Tensor& other) { in linalg_solve_is_vector_rhs() function
H A DBatchLinearAlgebra.cpp516 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 DFunctionsManual.cpp873 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.py1390 def linalg_solve_is_vector_rhs(input: Tensor, other: Tensor) -> bool: function
1418 vector_case = linalg_solve_is_vector_rhs(A, B)