Searched defs:vector_case (Results 1 – 3 of 3) sorted by relevance
543 …bool vector_case = other.dim() == 1 || (input.dim() - 1 == other.dim() && other.sym_sizes().equals… in linalg_solve_is_vector_rhs() local
516 const bool vector_case = at::native::linalg_solve_is_vector_rhs(A, B); in TORCH_META_FUNC() local1900 const bool vector_case = at::native::linalg_solve_is_vector_rhs(LU, B); in TORCH_IMPL_FUNC() local3358 bool vector_case = linalg_solve_is_vector_rhs(input, other); in linalg_lstsq_out_info() local3544 bool vector_case = linalg_solve_is_vector_rhs(input, other); in linalg_lstsq_out() local
3875 const bool vector_case = at::native::linalg_solve_is_vector_rhs(A, B_); in linalg_lstsq_backward() local5849 const bool vector_case = at::native::linalg_solve_is_vector_rhs(LU, X); in linalg_solve_jvp() local5888 const bool vector_case = at::native::linalg_solve_is_vector_rhs(LU, X); in linalg_solve_backward() local