Searched refs:vector_to_matrix (Results 1 – 1 of 1) sorted by relevance
3876 const auto vector_to_matrix = [vector_case](const Tensor& X) { in linalg_lstsq_backward() local3883 auto gX = vector_to_matrix(gX_); in linalg_lstsq_backward()3884 auto B = vector_to_matrix(B_); in linalg_lstsq_backward()5850 const auto vector_to_matrix = [vector_case](const Tensor& X) { in linalg_solve_jvp() local5860 auto X_ = vector_to_matrix(X); in linalg_solve_jvp()5861 auto dB_ = vector_to_matrix(dB); in linalg_solve_jvp()5889 const auto vector_to_matrix = [vector_case](const Tensor& X) { in linalg_solve_backward() local5900 gB_ = at::linalg_solve(A.mH(), vector_to_matrix(gX), left); in linalg_solve_backward()5904 LU, pivots, vector_to_matrix(gX), left, /*adjoint*/ !use_A_T); in linalg_solve_backward()5909 auto X_ = vector_to_matrix(X); in linalg_solve_backward()