Function faiss_sys::faiss_IndexIVF_search_preassigned [] [src]

pub unsafe extern "C" fn faiss_IndexIVF_search_preassigned(
    index: *const FaissIndexIVF,
    n: idx_t,
    x: *const f32,
    k: idx_t,
    assign: *const idx_t,
    centroid_dis: *const f32,
    distances: *mut f32,
    labels: *mut idx_t,
    store_pairs: c_int
) -> c_int

search a set of vectors, that are pre-quantized by the IVF quantizer. Fill in the corresponding heaps with the query results. search() calls this.

@param n nb of vectors to query @param x query vectors, size nx * d @param assign coarse quantization indices, size nx * nprobe @param centroid_dis distances to coarse centroids, size nx * nprobe @param distance output distances, size n * k @param labels output labels, size n * k @param store_pairs store inv list index + inv list offset instead in upper/lower 32 bit of result, instead of ids (used for reranking).