Searched defs:b_dim (Results 1 – 2 of 2) sorted by relevance
554 int b_dim = inv_permutation[a_dim]; in RemoveTrivialDimensions() local572 for (int b_dim = 0; b_dim < ndim; ++b_dim) { in RemoveTrivialDimensions() local635 for (int b_dim = 0; b_dim < ndim; ++b_dim) { in CoalesceDimensions() local768 int b_dim = inverse_permutation[a_dim]; in BuildPlanNodes() local1156 int b_dim = inverse_permutation[l.dim_in_a]; in Initialize() local1223 int b_dim = inverse_permutation[a_dim]; in ChooseParallelizationStrategy() local
1308 int b_dim = bigger_rank_shape->dim(b_i).size(); in CountBatchMatMulOperations() local