Searched refs:embedding_executor (Results 1 – 5 of 5) sorted by relevance
| /external/libtextclassifier/native/actions/ |
| D | feature-processor.cc | 83 const EmbeddingExecutor* embedding_executor, in AppendFeatures() argument
90 if (!embedding_executor->AddEmbedding( in AppendFeatures()
106 const Token& token, const EmbeddingExecutor* embedding_executor, in AppendTokenFeatures() argument
116 return AppendFeatures(sparse_features, dense_features, embedding_executor, in AppendTokenFeatures()
122 const EmbeddingExecutor* embedding_executor, in AppendTokenFeatures() argument
125 if (!AppendTokenFeatures(token, embedding_executor, output_features)) { in AppendTokenFeatures()
|
| D | feature-processor.h | 45 const EmbeddingExecutor* embedding_executor,
50 const EmbeddingExecutor* embedding_executor,
56 const EmbeddingExecutor* embedding_executor,
|
| /external/libtextclassifier/native/annotator/ |
| D | feature-processor.h | 172 const EmbeddingExecutor* embedding_executor,
289 const EmbeddingExecutor* embedding_executor,
|
| D | feature-processor.cc | 793 const EmbeddingExecutor* embedding_executor, in ExtractFeatures() argument
800 embedding_executor, embedding_cache, in ExtractFeatures()
811 embedding_executor, embedding_cache, in ExtractFeatures()
830 const EmbeddingExecutor* embedding_executor, in AppendTokenFeaturesWithCache() argument
871 if (!embedding_executor->AddEmbedding( in AppendTokenFeaturesWithCache()
|
| D | feature-processor_test.cc | 706 FakeEmbeddingExecutor embedding_executor; in TEST_F() local
713 /*selection_span_for_feature=*/{4, 11}, &embedding_executor, in TEST_F()
748 FakeEmbeddingExecutor embedding_executor; in TEST_F() local
768 &embedding_executor, &embedding_cache, /*feature_vector_size=*/4, in TEST_F()
|