Home
last modified time | relevance | path

Searched refs:vector_to_matrix (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp3876 const auto vector_to_matrix = [vector_case](const Tensor& X) { in linalg_lstsq_backward() local
3883 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() local
5860 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() local
5900 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()