Searched refs:b_dim (Results 1 – 1 of 1) sorted by relevance
814 int b_dim = bigger_rank_shape->dim(b_i).size(); in CountBatchMatMulOperations() local819 num_matmuls *= std::max(b_dim, s_dim); in CountBatchMatMulOperations()