Searched refs:embedding_cache (Results 1 – 5 of 5) sorted by relevance
| /external/libtextclassifier/ |
| D | feature-processor.cc | 792 EmbeddingCache* embedding_cache, int feature_vector_size, in ExtractFeatures() argument
798 embedding_executor, embedding_cache, in ExtractFeatures()
809 embedding_executor, embedding_cache, in ExtractFeatures()
936 EmbeddingCache* embedding_cache, in AppendTokenFeaturesWithCache() argument
939 if (embedding_cache) { in AppendTokenFeaturesWithCache()
940 const auto it = embedding_cache->find({token.start, token.end}); in AppendTokenFeaturesWithCache()
941 if (it != embedding_cache->end()) { in AppendTokenFeaturesWithCache()
987 if (embedding_cache) { in AppendTokenFeaturesWithCache()
988 (*embedding_cache)[{token.start, token.end}] = std::vector<float>( in AppendTokenFeaturesWithCache()
|
| D | feature-processor_test.cc | 693 FeatureProcessor::EmbeddingCache embedding_cache = { in TEST() local
702 &embedding_executor, &embedding_cache, /*feature_vector_size=*/4, in TEST()
716 EXPECT_EQ(embedding_cache.size(), 7); in TEST()
718 ElementsAreFloat(embedding_cache.at({0, 3}))); in TEST()
720 ElementsAreFloat(embedding_cache.at({8, 11}))); in TEST()
722 ElementsAreFloat(embedding_cache.at({8, 11}))); in TEST()
724 ElementsAreFloat(embedding_cache.at({16, 19}))); in TEST()
726 ElementsAreFloat(embedding_cache.at({20, 23}))); in TEST()
|
| D | feature-processor.h | 172 EmbeddingCache* embedding_cache, int feature_vector_size,
302 EmbeddingCache* embedding_cache,
|
| D | text-classifier.h | 232 FeatureProcessor::EmbeddingCache* embedding_cache,
238 FeatureProcessor::EmbeddingCache* embedding_cache,
|
| D | text-classifier.cc | 755 FeatureProcessor::EmbeddingCache* embedding_cache, in ModelClassifyText() argument
763 embedding_cache, classification_results); in ModelClassifyText()
821 FeatureProcessor::EmbeddingCache* embedding_cache, in ModelClassifyText() argument
890 embedding_cache, in ModelClassifyText()
1117 FeatureProcessor::EmbeddingCache embedding_cache; in ModelAnnotate() local
1167 interpreter_manager, &embedding_cache, in ModelAnnotate()
|