Searched refs:matrix_to_vector (Results 1 – 1 of 1) sorted by relevance
3879 const auto matrix_to_vector = [vector_case](const Tensor& X) { in linalg_lstsq_backward() local3896 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() local5865 return matrix_to_vector(dX_); in linalg_solve_jvp()5892 const auto matrix_to_vector = [vector_case](const Tensor& X) { in linalg_solve_backward() local5914 B_requires_grad ? matrix_to_vector(gB_) : Tensor{}); in linalg_solve_backward()