N2

stable

Template Class HnswSearchImpl¶

Defined in File hnsw_search_impl.h

Inheritance Relationships¶

Base Type¶

public n2::HnswSearch (Class HnswSearch)

Class Documentation¶

template<typename DistFuncType> class n2::HnswSearchImpl : public n2::HnswSearch ¶

Public Functions

HnswSearchImpl(std::shared_ptr<const HnswModel> model, size_t data_dim, DistanceKind metric)¶

void SearchByVector(const std::vector<float> &qvec, size_t k, int ef_search, bool ensure_k, std::vector<int> &result) override¶

void SearchByVector(const std::vector<float> &qvec, size_t k, int ef_search, bool ensure_k, std::vector<std::pair<int, float>> &result) override¶

void SearchById(int id, size_t k, int ef_search, bool ensure_k, std::vector<int> &result) override¶

void SearchById(int id, size_t k, int ef_search, bool ensure_k, std::vector<std::pair<int, float>> &result) override¶

Protected Functions

template<typename ResultType> void SearchByVector_(const std::vector<float> &qvec, size_t k, int ef_search, bool ensure_k, ResultType &result)¶

void CallSearchById_(int cur_node_id, float cur_dist, const float *qraw, size_t k, size_t ef_search, bool ensure_k, std::vector<int> &result)¶

void CallSearchById_(int cur_node_id, float cur_dist, const float *qraw, size_t k, size_t ef_search, bool ensure_k, std::vector<std::pair<int, float>> &result)¶

template<typename ResultType> void SearchById_(int cur_node_id, float cur_dist, const float *qraw, size_t k, size_t ef_search, bool ensure_k, ResultType &result)¶

template<typename ResultType> void SearchByIdV1_(int cur_node_id, float cur_dist, const float *qraw, size_t k, size_t ef_search, bool ensure_k, ResultType &result)¶

template<typename ResultType> void SearchByIdV2_(int cur_node_id, float cur_dist, const float *qraw, size_t k, size_t ef_search, bool ensure_k, ResultType &result)¶

bool PrepareEnsureKSearch(int cur_node_id, std::vector<int> &result, IdDistancePairMinHeap &visited_nodes)¶

bool PrepareEnsureKSearch(int cur_node_id, std::vector<std::pair<int, float>> &result, IdDistancePairMinHeap &visited_nodes)¶

void MakeSearchResult(size_t k, IdDistancePairMinHeap &candidates, IdDistancePairMinHeap &visited_nodes, std::vector<int> &result)¶

void MakeSearchResult(size_t k, IdDistancePairMinHeap &candidates, IdDistancePairMinHeap &visited_nodes, std::vector<std::pair<int, float>> &result)¶

Protected Attributes

std::shared_ptr<const HnswModel> model_¶

std::unique_ptr<VisitedList> visited_list_¶

size_t data_dim_¶

DistanceKind metric_¶

DistFuncType dist_func_¶

std::vector<float> normalized_vec_¶

std::vector<std::pair<int, float>> ensure_k_path_¶

char *model_higher_level_ = nullptr¶

char *model_level0_ = nullptr¶

char *model_level0_node_base_offset_ = nullptr¶

uint64_t memory_per_node_level0_¶

uint64_t memory_per_node_higher_level_¶