Open3D (C++ API)  0.18.0+252c867
Macros | Functions
KnnSearchOpKernel.cpp File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

#include "open3d/core/nns/NanoFlannImpl.h"
#include "open3d/core/nns/NeighborSearchCommon.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/NeighborSearchAllocator.h"
#include "torch/script.h"

Macros

#define INSTANTIATE(T, TIndex)
 

Functions

template<class T , class TIndex >
void KnnSearchCPU (const torch::Tensor &points, const torch::Tensor &queries, const int64_t k, const torch::Tensor &points_row_splits, const torch::Tensor &queries_row_splits, const Metric metric, const bool ignore_query_point, const bool return_distances, torch::Tensor &neighbors_index, torch::Tensor &neighbors_row_splits, torch::Tensor &neighbors_distance)
 

Macro Definition Documentation

◆ INSTANTIATE

#define INSTANTIATE (   T,
  TIndex 
)
Value:
template void KnnSearchCPU<T, TIndex>( \
const torch::Tensor& points, const torch::Tensor& queries, \
const int64_t k, const torch::Tensor& points_row_splits, \
const torch::Tensor& queries_row_splits, const Metric metric, \
const bool ignore_query_point, const bool return_distances, \
torch::Tensor& neighbors_index, \
torch::Tensor& neighbors_row_splits, \
torch::Tensor& neighbors_distance);
int points
Definition: FilePCD.cpp:54
Metric
Supported metrics.
Definition: NeighborSearchCommon.h:19

Function Documentation

◆ KnnSearchCPU()

template<class T , class TIndex >
void KnnSearchCPU ( const torch::Tensor &  points,
const torch::Tensor &  queries,
const int64_t  k,
const torch::Tensor &  points_row_splits,
const torch::Tensor &  queries_row_splits,
const Metric  metric,
const bool  ignore_query_point,
const bool  return_distances,
torch::Tensor &  neighbors_index,
torch::Tensor &  neighbors_row_splits,
torch::Tensor &  neighbors_distance 
)