20#ifdef BUILD_CUDA_MODULE
21template <
class T,
class TIndex>
22void KnnSearchCUDA(
const Tensor&
points,
23 const Tensor& points_row_splits,
24 const Tensor& queries,
25 const Tensor& queries_row_splits,
27 Tensor& neighbors_index,
28 Tensor& neighbors_row_splits,
29 Tensor& neighbors_distance);
50 const Tensor& points_row_splits,
56 "[KnnIndex::SetTensorData with radius not implemented.");
60 int knn)
const override;
63 const Tensor& queries_row_splits,
68 bool sort)
const override {
69 utility::LogError(
"KnnIndex::SearchRadius not implemented.");
74 bool sort)
const override {
75 utility::LogError(
"KnnIndex::SearchRadius not implemented.");
79 const Tensor& query_points,
81 const int max_knn)
const override {
82 utility::LogError(
"KnnIndex::SearchHybrid not implemented.");
std::pair< Tensor, Tensor > SearchKnn(const Tensor &query_points, int knn) const override
Definition KnnIndex.cpp:79
std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, const double radius, bool sort) const override
Definition KnnIndex.h:72
KnnIndex(const KnnIndex &)=delete
KnnIndex()
Definition KnnIndex.cpp:19
std::tuple< Tensor, Tensor, Tensor > SearchHybrid(const Tensor &query_points, const double radius, const int max_knn) const override
Definition KnnIndex.h:78
bool SetTensorData(const Tensor &dataset_points, const Dtype &index_dtype=core::Int64) override
Definition KnnIndex.cpp:31
KnnIndex & operator=(const KnnIndex &)=delete
Tensor points_row_splits_
Definition KnnIndex.h:86
~KnnIndex()
Definition KnnIndex.cpp:29
bool SetTensorData(const Tensor &dataset_points, double radius, const Dtype &index_dtype=core::Int64) override
Definition KnnIndex.h:52
std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, const Tensor &radii, bool sort) const override
Definition KnnIndex.h:66
const Dtype Int64
Definition Dtype.cpp:47
Definition PinholeCameraIntrinsic.cpp:16