39 class StandardGpuResources;
64 bool SetTensorData(
const Tensor &dataset_points)
override;
68 "FaissIndex::SetTensorData with radius not implemented.");
72 std::pair<Tensor, Tensor> SearchKnn(
const Tensor &query_points,
73 int knn)
const override;
76 const Tensor &query_points,
const Tensor &radii)
const override {
81 const Tensor &query_points,
double radius)
const override {
86 std::pair<Tensor, Tensor> SearchHybrid(
const Tensor &query_points,
88 int max_knn)
const override;
91 std::unique_ptr<faiss::Index>
index;
92 #ifdef BUILD_CUDA_MODULE 93 std::unique_ptr<faiss::gpu::StandardGpuResources> res;
std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, double radius) const override
Definition: FaissIndex.h:80
Definition: FaissIndex.h:36
Definition: NNSIndex.h:40
void LogError(const char *format, const Args &... args)
Definition: Console.h:176
std::unique_ptr< faiss::Index > index
Definition: FaissIndex.h:91
bool SetTensorData(const Tensor &dataset_points, double radius) override
Definition: FaissIndex.h:66
Definition: PinholeCameraIntrinsic.cpp:35
std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, const Tensor &radii) const override
Definition: FaissIndex.h:75
Faiss for nearest neighbor search.
Definition: FaissIndex.h:50