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.cc1431 FeatureProcessor::EmbeddingCache* embedding_cache, in ModelClassifyText() argument
1441 interpreter_manager, embedding_cache, in ModelClassifyText()
1452 FeatureProcessor::EmbeddingCache* embedding_cache, in ModelClassifyText() argument
1537 embedding_executor_.get(), embedding_cache, in ModelClassifyText()
2045 FeatureProcessor::EmbeddingCache embedding_cache; in ModelAnnotate() local
2070 codepoint_span, options, interpreter_manager, &embedding_cache, in ModelAnnotate()