Searched refs:smaller_rank_shape (Results 1 – 1 of 1) sorted by relevance
805 TensorShapeProto* smaller_rank_shape = &b_input_shape; in CountBatchMatMulOperations() local808 smaller_rank_shape = &a_input_shape; in CountBatchMatMulOperations()812 s_i = smaller_rank_shape->dim_size() - bigger_rank_shape->dim_size(); in CountBatchMatMulOperations()817 s_dim = smaller_rank_shape->dim(s_i).size(); in CountBatchMatMulOperations()