Lines Matching refs:classification_feature_processor_

209     classification_feature_processor_.reset(new FeatureProcessor(  in ValidateAndInitialize()
802           classification_feature_processor_->GetOptions()  in ClassifyTextUpperBoundNeededTokens()
825     tokens = classification_feature_processor_->Tokenize(context);  in ModelClassifyText()
832   classification_feature_processor_->RetokenizeAndFindClick(  in ModelClassifyText()
834       classification_feature_processor_->GetOptions()  in ModelClassifyText()
849           classification_feature_processor_->GetOptions()  in ModelClassifyText()
874         classification_feature_processor_->GetOptions()->context_size();  in ModelClassifyText()
881   if (!classification_feature_processor_->HasEnoughSupportedCodepoints(  in ModelClassifyText()
888   if (!classification_feature_processor_->ExtractFeatures(  in ModelClassifyText()
891           classification_feature_processor_->EmbeddingSize() +  in ModelClassifyText()
892               classification_feature_processor_->DenseFeaturesCount(),  in ModelClassifyText()
917       logits.dim(1) != classification_feature_processor_->NumCollections()) {  in ModelClassifyText()
928         classification_feature_processor_->LabelToCollection(i), scores[i]};  in ModelClassifyText()
1197   return classification_feature_processor_.get();  in ClassificationFeatureProcessorForTests()