Searched defs:input_batch_shape (Results 1 – 1 of 1) sorted by relevance
3408 auto input_batch_shape = IntArrayRef(input.sizes().cbegin(), input.sizes().cend() - 2); in linalg_lstsq_out_info() local3608 auto input_batch_shape = IntArrayRef(input.sizes().cbegin(), input.sizes().cend() - 2); in linalg_lstsq_out() local