Home
last modified time | relevance | path

Searched refs:embedding_executor_ (Results 1 – 6 of 6) sorted by relevance

/external/libtextclassifier/actions/
Dfeature-processor_test.cc61 FakeEmbeddingExecutor embedding_executor_; member in libtextclassifier3::__anona6cf22cd0111::FeatureProcessorTest
79 EXPECT_TRUE(feature_processor.AppendTokenFeatures(token, &embedding_executor_, in TEST_F()
99 EXPECT_TRUE(feature_processor.AppendTokenFeatures(token, &embedding_executor_, in TEST_F()
122 tokens, &embedding_executor_, &token_features)); in TEST_F()
Dactions-suggestions.cc316 embedding_executor_ = TFLiteEmbeddingExecutor::FromBuffer( in ValidateAndInitialize()
320 if (embedding_executor_ == nullptr) { in ValidateAndInitialize()
417 /*dense_features=*/{}, embedding_executor_.get(), embedding); in EmbedTokenId()
633 tokens[i][pos], embedding_executor_.get(), embeddings)) { in EmbedTokensPerMessage()
689 tokens[i][pos], embedding_executor_.get(), embeddings)) { in EmbedAndFlattenTokens()
Dactions-suggestions.h149 std::unique_ptr<const EmbeddingExecutor> embedding_executor_; variable
Dactions-suggestions_test.cc1053 embedding_executor_.reset(new FakeEmbeddingExecutor()); in TestingMessageEmbedder()
/external/libtextclassifier/annotator/
Dannotator.cc352 embedding_executor_ = TFLiteEmbeddingExecutor::FromBuffer( in ValidateAndInitialize()
357 if (!embedding_executor_) { in ValidateAndInitialize()
1079 embedding_executor_.get(), in ModelSuggestSelection()
1285 embedding_executor_.get(), embedding_cache, in ModelClassifyText()
1686 embedding_executor_.get(), in ModelAnnotate()
Dannotator.h466 std::unique_ptr<const EmbeddingExecutor> embedding_executor_; variable