Searched refs:smaller_rank_shape (Results 1 – 1 of 1) sorted by relevance
1288 TensorShapeProto* smaller_rank_shape = &b_input_shape; in CountBatchMatMulOperations() local1291 smaller_rank_shape = &a_input_shape; in CountBatchMatMulOperations()1295 s_i = smaller_rank_shape->dim_size() - bigger_rank_shape->dim_size(); in CountBatchMatMulOperations()1300 s_dim = smaller_rank_shape->dim(s_i).size(); in CountBatchMatMulOperations()