Searched refs:bigger_rank_shape (Results 1 – 1 of 1) sorted by relevance
1287 TensorShapeProto* bigger_rank_shape = &a_input_shape; in CountBatchMatMulOperations() local1290 bigger_rank_shape = &b_input_shape; in CountBatchMatMulOperations()1295 s_i = smaller_rank_shape->dim_size() - bigger_rank_shape->dim_size(); in CountBatchMatMulOperations()1296 b_i < bigger_rank_shape->dim_size() - matrix_rank; ++b_i, ++s_i) { in CountBatchMatMulOperations()1297 int b_dim = bigger_rank_shape->dim(b_i).size(); in CountBatchMatMulOperations()