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