Searched refs:bigger_rank_shape (Results 1 – 1 of 1) sorted by relevance
804 TensorShapeProto* bigger_rank_shape = &a_input_shape; in CountBatchMatMulOperations() local807 bigger_rank_shape = &b_input_shape; in CountBatchMatMulOperations()812 s_i = smaller_rank_shape->dim_size() - bigger_rank_shape->dim_size(); in CountBatchMatMulOperations()813 b_i < bigger_rank_shape->dim_size() - matrix_rank; ++b_i, ++s_i) { in CountBatchMatMulOperations()814 int b_dim = bigger_rank_shape->dim(b_i).size(); in CountBatchMatMulOperations()