Searched defs:actual_batch_tau_shape (Results 1 – 1 of 1) sorted by relevance
2512 auto actual_batch_tau_shape = IntArrayRef(tau.sizes().data(), tau.dim() - 1); // tau.shape[:-1] in linalg_householder_product_out() local2664 auto actual_batch_tau_shape = IntArrayRef(tau.sizes().data(), tau.dim() - 1); // tau.shape[:-1] in ormqr_out() local