Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp3879 const auto matrix_to_vector = [vector_case](const Tensor& X) { in linalg_lstsq_backward() local
3896 B_grad = matrix_to_vector(pinvA.mH().matmul(gX)); in linalg_lstsq_backward()
5853 const auto matrix_to_vector = [vector_case](const Tensor& X) { in linalg_solve_jvp() local
5865 return matrix_to_vector(dX_); in linalg_solve_jvp()
5892 const auto matrix_to_vector = [vector_case](const Tensor& X) { in linalg_solve_backward() local
5914 B_requires_grad ? matrix_to_vector(gB_) : Tensor{}); in linalg_solve_backward()