Searched defs:expected_batch_tau_shape (Results 1 – 1 of 1) sorted by relevance
2276 …auto expected_batch_tau_shape = IntArrayRef(input.sizes().data(), input.dim() - 2).vec(); // input… in geqrf_out_helper() local2307 …auto expected_batch_tau_shape = IntArrayRef(input.sizes().data(), input.dim() - 2).vec(); // input… in geqrf_out() local2511 …auto expected_batch_tau_shape = IntArrayRef(input.sizes().data(), input.dim() - 2); // input.shape… in linalg_householder_product_out() local