Searched defs:b_batch_shape (Results 1 – 2 of 2) sorted by relevance
1522 IntArrayRef b_batch_shape(B.sizes().data(), B.dim() - 2); in lu_solve_looped_cusolver() local
330 ShapeHandle b_batch_shape; in BatchMatMulV2Shape() local