Struct faiss::ReservoirResultHandler
-
template<class C>
struct ReservoirResultHandler -
Public Functions
-
inline void begin_multiple(size_t i0, size_t i1)
begin
-
inline void add_results(size_t j0, size_t j1, const T *dis_tab)
add results for query i0..i1 and j0..j1
-
inline void end_multiple()
series of results for queries i0..i1 is done
Public Members
-
int nq
-
T *heap_dis_tab
-
TI *heap_ids_tab
-
int64_t k
-
size_t capacity
-
size_t i0
-
size_t i1
-
std::vector<ReservoirTopN<C>> reservoirs
-
struct SingleResultHandler
Public Functions
-
inline SingleResultHandler(ReservoirResultHandler &hr)
-
inline void begin(size_t i)
begin results for query # i
-
inline void end()
series of results for query i is done
-
inline SingleResultHandler(ReservoirResultHandler &hr)
-
inline void begin_multiple(size_t i0, size_t i1)