Home
last modified time | relevance | path

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

/external/libtextclassifier/native/actions/
Dfeature-processor_test.cc61 FakeEmbeddingExecutor embedding_executor_; member in libtextclassifier3::__anonbe8aa34a0111::ActionsFeatureProcessorTest
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.cc372 embedding_executor_ = TFLiteEmbeddingExecutor::FromBuffer( in ValidateAndInitialize()
376 if (embedding_executor_ == nullptr) { in ValidateAndInitialize()
471 /*dense_features=*/{}, embedding_executor_.get(), embedding); in EmbedTokenId()
517 tokens[i][pos], embedding_executor_.get(), embeddings)) { in EmbedTokensPerMessage()
573 tokens[i][pos], embedding_executor_.get(), embeddings)) { in EmbedAndFlattenTokens()
Dactions-suggestions.h141 std::unique_ptr<const EmbeddingExecutor> embedding_executor_; variable
Dactions-suggestions_test.cc1404 embedding_executor_.reset(new FakeEmbeddingExecutor()); in TestingMessageEmbedder()
/external/libtextclassifier/native/annotator/
Dannotator.cc402 embedding_executor_ = TFLiteEmbeddingExecutor::FromBuffer( in ValidateAndInitialize()
407 if (!embedding_executor_) { in ValidateAndInitialize()
1320 embedding_executor_.get(), in ModelSuggestSelection()
1529 embedding_executor_.get(), embedding_cache, in ModelClassifyText()
2002 embedding_executor_.get(), in ModelAnnotate()
Dannotator.h447 std::unique_ptr<const EmbeddingExecutor> embedding_executor_; variable