Searched refs:DenseFeaturesCount (Results 1 – 4 of 4) sorted by relevance
154 int DenseFeaturesCount() const { in DenseFeaturesCount() function155 return feature_extractor_.DenseFeaturesCount(); in DenseFeaturesCount()
75 int DenseFeaturesCount() const { in DenseFeaturesCount() function
246 embedding_size + feature_processor.DenseFeaturesCount(), &tokens, in InferInternal()378 embedding_size + selection_feature_processor_->DenseFeaturesCount(), in SuggestSelectionSymmetrical()
653 feature_extractor_.DenseFeaturesCount()); in ExtractFeatures()