Searched refs:matrix_rank (Results 1 – 1 of 1) sorted by relevance
785 const int matrix_rank = 2; in CountBatchMatMulOperations() local791 a_input.shape(), std::max(matrix_rank, a_input.shape().dim_size()), in CountBatchMatMulOperations()794 b_input.shape(), std::max(matrix_rank, b_input.shape().dim_size()), in CountBatchMatMulOperations()798 (a_input.shape().dim_size() < matrix_rank) || in CountBatchMatMulOperations()799 (b_input.shape().dim_size() < matrix_rank); in CountBatchMatMulOperations()813 b_i < bigger_rank_shape->dim_size() - matrix_rank; ++b_i, ++s_i) { in CountBatchMatMulOperations()844 for (int i = std::max(0, a_input_shape.dim_size() - matrix_rank); in CountBatchMatMulOperations()852 for (int i = std::max(0, b_input_shape.dim_size() - matrix_rank); in CountBatchMatMulOperations()