Home
last modified time | relevance | path

Searched refs:embedding_cache (Results 1 – 5 of 5) sorted by relevance

/external/libtextclassifier/native/annotator/
Dfeature-processor.cc794 EmbeddingCache* embedding_cache, int feature_vector_size, in ExtractFeatures() argument
800 embedding_executor, embedding_cache, in ExtractFeatures()
811 embedding_executor, embedding_cache, in ExtractFeatures()
831 EmbeddingCache* embedding_cache, in AppendTokenFeaturesWithCache() argument
834 if (embedding_cache) { in AppendTokenFeaturesWithCache()
835 const auto it = embedding_cache->find({token.start, token.end}); in AppendTokenFeaturesWithCache()
836 if (it != embedding_cache->end()) { in AppendTokenFeaturesWithCache()
882 if (embedding_cache) { in AppendTokenFeaturesWithCache()
883 (*embedding_cache)[{token.start, token.end}] = std::vector<float>( in AppendTokenFeaturesWithCache()
Dfeature-processor_test.cc759 FeatureProcessor::EmbeddingCache embedding_cache = { in TEST_F() local
768 &embedding_executor, &embedding_cache, /*feature_vector_size=*/4, in TEST_F()
782 EXPECT_EQ(embedding_cache.size(), 7); in TEST_F()
784 ElementsAreFloat(embedding_cache.at({0, 3}))); in TEST_F()
786 ElementsAreFloat(embedding_cache.at({8, 11}))); in TEST_F()
788 ElementsAreFloat(embedding_cache.at({8, 11}))); in TEST_F()
790 ElementsAreFloat(embedding_cache.at({16, 19}))); in TEST_F()
792 ElementsAreFloat(embedding_cache.at({20, 23}))); in TEST_F()
Dfeature-processor.h173 EmbeddingCache* embedding_cache, int feature_vector_size,
290 EmbeddingCache* embedding_cache,
Dannotator.h306 FeatureProcessor::EmbeddingCache* embedding_cache,
324 FeatureProcessor::EmbeddingCache* embedding_cache,
Dannotator.cc1423 FeatureProcessor::EmbeddingCache* embedding_cache, in ModelClassifyText() argument
1433 interpreter_manager, embedding_cache, in ModelClassifyText()
1444 FeatureProcessor::EmbeddingCache* embedding_cache, in ModelClassifyText() argument
1529 embedding_executor_.get(), embedding_cache, in ModelClassifyText()
1977 FeatureProcessor::EmbeddingCache embedding_cache; in ModelAnnotate() local
2066 &embedding_cache, &classification, /*tokens=*/nullptr)) { in ModelAnnotate()
2075 options, interpreter_manager, &embedding_cache, in ModelAnnotate()