Searched refs:embedding_executor (Results 1 – 5 of 5) sorted by relevance
| /external/libtextclassifier/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/annotator/ |
| D | feature-processor.h | 153 const EmbeddingExecutor* embedding_executor,
260 const EmbeddingExecutor* embedding_executor,
|
| D | feature-processor.cc | 763 const EmbeddingExecutor* embedding_executor, in ExtractFeatures() argument
770 embedding_executor, embedding_cache, in ExtractFeatures()
781 embedding_executor, embedding_cache, in ExtractFeatures()
800 const EmbeddingExecutor* embedding_executor, in AppendTokenFeaturesWithCache() argument
841 if (!embedding_executor->AddEmbedding( in AppendTokenFeaturesWithCache()
|
| D | feature-processor_test.cc | 631 FakeEmbeddingExecutor embedding_executor; in TEST_F() local
638 /*selection_span_for_feature=*/{4, 11}, &embedding_executor, in TEST_F()
673 FakeEmbeddingExecutor embedding_executor; in TEST_F() local
693 &embedding_executor, &embedding_cache, /*feature_vector_size=*/4, in TEST_F()
|