|
|
| GpuIndexIVF (GpuResources *resources, int device, IndicesOptions indicesOptions, bool useFloat16CoarseQuantizer, int dims, faiss::MetricType metric, int nlist) |
| |
|
| GpuIndexIVF (GpuResources *resources, int device, IndicesOptions indicesOptions, int dims, faiss::MetricType metric, int nlist, GpuIndexFlat *quantizer) |
| |
|
IndicesOptions | getIndicesOptions () const |
| | What indices storage options are we using?
|
| |
|
bool | getUseFloat16CoarseQuantizer () const |
| | Is our coarse quantizer storing and performing math in float16?
|
| |
|
void | copyFrom (const faiss::IndexIVF *index) |
| | Copy what we need from the CPU equivalent.
|
| |
|
void | copyTo (faiss::IndexIVF *index) const |
| | Copy what we have to the CPU equivalent.
|
| |
|
int | getNumLists () const |
| | Returns the number of inverted lists we're managing.
|
| |
|
void | setNumProbes (int nprobe) |
| | Sets the number of list probes per query.
|
| |
|
int | getNumProbes () const |
| | Returns our current number of list probes per query.
|
| |
| void | add (Index::idx_t n, const float *x) override |
| |
|
| GpuIndex (GpuResources *resources, int device, int dims, faiss::MetricType metric) |
| |
|
int | getDevice () const |
| |
|
GpuResources * | getResources () |
| |
|
| Index (idx_t d=0, MetricType metric=METRIC_INNER_PRODUCT) |
| |
| virtual void | train (idx_t n, const float *x) |
| |
| virtual void | add_with_ids (idx_t n, const float *x, const long *xids) |
| |
| virtual void | search (idx_t n, const float *x, idx_t k, float *distances, idx_t *labels) const =0 |
| |
| virtual void | range_search (idx_t n, const float *x, float radius, RangeSearchResult *result) const |
| |
| void | assign (idx_t n, const float *x, idx_t *labels, idx_t k=1) |
| |
|
virtual void | reset ()=0 |
| | removes all elements from the database.
|
| |
| virtual long | remove_ids (const IDSelector &sel) |
| |
| virtual void | reconstruct (idx_t key, float *recons) const |
| |
| virtual void | reconstruct_n (idx_t i0, idx_t ni, float *recons) const |
| |
| void | compute_residual (const float *x, float *residual, idx_t key) const |
| |
| void | display () const |
| |
| virtual std::string | get_typename () const |
| |
|
virtual void | set_typename ()=0 |
| |
Definition at line 25 of file GpuIndexIVF.h.