Searched refs:NumCollections (Results 1 – 2 of 2) sorted by relevance
124 int NumCollections() const { return collection_to_label_.size(); } in NumCollections() function
1312 logits.dim(1) != classification_feature_processor_->NumCollections()) { in ModelClassifyText()