Home
last modified time | relevance | path

Searched defs:expected_solution_shape (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DBatchLinearAlgebra.cpp3364 …std::vector<int64_t> expected_solution_shape = broadcast_batch_size(input, other_2d, input.dim() -… in linalg_lstsq_out_info() local
3588 …std::vector<int64_t> expected_solution_shape = broadcast_batch_size(input, other_2d, input.dim() -… in linalg_lstsq_out() local