Namespace faiss::rq_encode_steps

namespace rq_encode_steps

Functions

void refine_beam_mp(const ResidualQuantizer &rq, size_t n, size_t beam_size, const float *x, int out_beam_size, int32_t *out_codes, float *out_residuals, float *out_distances, RefineBeamMemoryPool &pool)

void refine_beam_LUT_mp(const ResidualQuantizer &rq, size_t n, const float *query_norms, const float *query_cp, int out_beam_size, int32_t *out_codes, float *out_distances, RefineBeamLUTMemoryPool &pool)

void compute_codes_add_centroids_mp_lut0(const ResidualQuantizer &rq, const float *x, uint8_t *codes_out, size_t n, const float *centroids, ComputeCodesAddCentroidsLUT0MemoryPool &pool)

void compute_codes_add_centroids_mp_lut1(const ResidualQuantizer &rq, const float *x, uint8_t *codes_out, size_t n, const float *centroids, ComputeCodesAddCentroidsLUT1MemoryPool &pool)

struct RefineBeamMemoryPool

Public Members

std::vector<int32_t> new_codes

std::vector<float> new_residuals

std::vector<float> residuals

std::vector<int32_t> codes

std::vector<float> distances

struct RefineBeamLUTMemoryPool

Public Members

std::vector<int32_t> new_codes

std::vector<float> new_distances

std::vector<int32_t> codes

std::vector<float> distances

struct ComputeCodesAddCentroidsLUT0MemoryPool

Public Members

std::vector<int32_t> codes

std::vector<float> norms

std::vector<float> distances

std::vector<float> residuals

RefineBeamMemoryPool refine_beam_pool

struct ComputeCodesAddCentroidsLUT1MemoryPool

Public Members

std::vector<int32_t> codes

std::vector<float> distances

std::vector<float> query_norms

std::vector<float> query_cp

std::vector<float> residuals

RefineBeamLUTMemoryPool refine_beam_lut_pool