| /external/icing/icing/testing/ |
| D | embedding-test-utils.cc | 36 GetEmbeddingHitsFromIndex(const EmbeddingIndex* embedding_index, in GetEmbeddingHitsFromIndex() argument
42 pl_accessor_or = embedding_index->GetAccessor(dimension, model_signature); in GetEmbeddingHitsFromIndex()
61 const EmbeddingIndex* embedding_index) { in GetRawEmbeddingDataFromIndex() argument
63 embedding_index->GetRawEmbeddingData(), in GetRawEmbeddingDataFromIndex()
65 return std::vector<float>(data, data + embedding_index->GetTotalVectorSize()); in GetRawEmbeddingDataFromIndex()
70 const EmbeddingIndex* embedding_index, const EmbeddingHit& hit, in GetAndRestoreQuantizedEmbeddingVectorFromIndex() argument
74 embedding_index->GetQuantizedEmbeddingVector(hit, dimension)); in GetAndRestoreQuantizedEmbeddingVectorFromIndex()
|
| D | embedding-test-utils.h | 49 GetEmbeddingHitsFromIndex(const EmbeddingIndex* embedding_index,
53 const EmbeddingIndex* embedding_index);
59 const EmbeddingIndex* embedding_index, const EmbeddingHit& hit,
|
| /external/icing/icing/index/embed/ |
| D | embedding-index_test.cc | 185 std::unique_ptr<EmbeddingIndex> embedding_index, in TEST_F()
190 ICING_ASSERT_OK(embedding_index->BufferEmbedding( in TEST_F()
193 ICING_ASSERT_OK(embedding_index->CommitBufferToIndex()); in TEST_F()
194 embedding_index->set_last_added_document_id(0); in TEST_F()
197 GetEmbeddingHitsFromIndex(embedding_index.get(), /*dimension=*/3, in TEST_F()
201 EXPECT_THAT(GetRawEmbeddingDataFromIndex(embedding_index.get()), in TEST_F()
203 EXPECT_EQ(embedding_index->last_added_document_id(), 0); in TEST_F()
205 ICING_EXPECT_OK(embedding_index->GetChecksum()); in TEST_F()
213 embedding_index.reset(); in TEST_F()
222 std::unique_ptr<EmbeddingIndex> embedding_index, in TEST_F()
[all …]
|
| D | doc-hit-info-iterator-embedding.cc | 49 const EmbeddingIndex* embedding_index, const DocumentStore* document_store, in Create() argument
52 ICING_RETURN_ERROR_IF_NULL(embedding_index); in Create()
58 pl_accessor_or = embedding_index->GetAccessorForVector(*query); in Create()
77 score_high, get_embedding_match_info, info_map, embedding_index, in Create()
|
| D | doc-hit-info-iterator-embedding.h | 66 const EmbeddingIndex* embedding_index,
100 const EmbeddingIndex* embedding_index, in DocHitInfoIteratorEmbedding() argument
111 embedding_index_(*embedding_index), in DocHitInfoIteratorEmbedding()
|
| /external/icing/icing/index/ |
| D | embedding-indexing-handler.h | 43 Create(const Clock* clock, EmbeddingIndex* embedding_index,
66 EmbeddingIndex* embedding_index, in EmbeddingIndexingHandler() argument
69 embedding_index_(*embedding_index), in EmbeddingIndexingHandler()
|
| D | embedding-indexing-handler.cc | 36 EmbeddingIndex* embedding_index, in Create() argument
39 ICING_RETURN_ERROR_IF_NULL(embedding_index); in Create()
42 clock, embedding_index, enable_embedding_index)); in Create()
|
| /external/icing/icing/query/ |
| D | query-processor.cc | 60 const EmbeddingIndex* embedding_index, in Create() argument
69 ICING_RETURN_ERROR_IF_NULL(embedding_index); in Create()
78 index, numeric_index, embedding_index, language_segmenter, normalizer, in Create()
85 const EmbeddingIndex* embedding_index, in QueryProcessor() argument
92 embedding_index_(*embedding_index), in QueryProcessor()
|
| D | suggestion-processor.h | 52 const EmbeddingIndex* embedding_index,
70 const EmbeddingIndex* embedding_index,
|
| D | suggestion-processor.cc | 56 const EmbeddingIndex* embedding_index, in Create() argument
64 ICING_RETURN_ERROR_IF_NULL(embedding_index); in Create()
73 index, numeric_index, embedding_index, language_segmenter, normalizer, in Create()
346 const EmbeddingIndex* embedding_index, in SuggestionProcessor() argument
352 embedding_index_(*embedding_index), in SuggestionProcessor()
|
| D | query-processor.h | 53 const EmbeddingIndex* embedding_index,
82 const EmbeddingIndex* embedding_index,
|
| D | query-processor_benchmark.cc | 179 auto embedding_index, in BM_QueryOneTerm()
198 index.get(), numeric_index.get(), embedding_index.get(), in BM_QueryOneTerm()
318 auto embedding_index, in BM_QueryFiveTerms()
351 index.get(), numeric_index.get(), embedding_index.get(), in BM_QueryFiveTerms()
475 auto embedding_index, in BM_QueryDiacriticTerm()
497 index.get(), numeric_index.get(), embedding_index.get(), in BM_QueryDiacriticTerm()
617 auto embedding_index, in BM_QueryHiragana()
639 index.get(), numeric_index.get(), embedding_index.get(), in BM_QueryHiragana()
|
| /external/icing/icing/query/advanced_query_parser/ |
| D | query-visitor.h | 58 const EmbeddingIndex* embedding_index, in QueryVisitor() argument
66 : QueryVisitor(index, numeric_index, embedding_index, document_store, in QueryVisitor()
117 const EmbeddingIndex* embedding_index, in QueryVisitor() argument
129 embedding_index_(*embedding_index), in QueryVisitor()
|
| /external/icing/icing/scoring/advanced_scoring/ |
| D | score-expression.cc | 772 uint32_t embedding_index, in Create()
775 if (embedding_query_results->GetMatchInfoMap(embedding_index, in Create()
778 "The embedding query index ", std::to_string(embedding_index), in Create()
|
| /external/icing/icing/ |
| D | icing-search-engine_initialization_test.cc | 6134 std::unique_ptr<EmbeddingIndex> embedding_index, in TEST_P()
6154 EmbeddingIndexingHandler::Create(&fake_clock, embedding_index.get(), in TEST_P()
|