39 class StandardGpuResources;
69 "FaissIndex::SetTensorData with radius not implemented.");
81 int knn)
const override;
95 bool sort)
const override {
111 bool sort)
const override {
129 int max_knn)
const override;
132 std::unique_ptr<faiss::Index>
index;
133 #ifdef BUILD_CUDA_MODULE
134 std::unique_ptr<faiss::gpu::StandardGpuResources> res;
#define LogError(...)
Definition: Logging.h:72
Faiss for nearest neighbor search.
Definition: FaissIndex.h:50
~FaissIndex()
Definition: FaissIndex.cpp:56
std::tuple< Tensor, Tensor, Tensor > SearchHybrid(const Tensor &query_points, double radius, int max_knn) const override
Definition: FaissIndex.cpp:128
FaissIndex(const FaissIndex &)=delete
FaissIndex & operator=(const FaissIndex &)=delete
bool SetTensorData(const Tensor &dataset_points, double radius) override
Definition: FaissIndex.h:67
std::pair< Tensor, Tensor > SearchKnn(const Tensor &query_points, int knn) const override
Definition: FaissIndex.cpp:97
std::unique_ptr< faiss::Index > index
Definition: FaissIndex.h:132
std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, const Tensor &radii, bool sort) const override
Definition: FaissIndex.h:93
bool SetTensorData(const Tensor &dataset_points) override
Definition: FaissIndex.cpp:58
std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, double radius, bool sort) const override
Definition: FaissIndex.h:109
FaissIndex()
Default Constructor.
Definition: FaissIndex.cpp:49
Definition: NNSIndex.h:40
Definition: FaissIndex.h:36
Definition: PinholeCameraIntrinsic.cpp:35