KnnSearchOpKernelCPU< T, TIndex > Class Template Reference

Inheritance diagram for KnnSearchOpKernelCPU< T, TIndex >:

Public Member Functions
	KnnSearchOpKernelCPU (OpKernelConstruction *construction)
void	Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &points, const tensorflow::Tensor &queries, const int k, const tensorflow::Tensor &points_row_splits, const tensorflow::Tensor &queries_row_splits, tensorflow::Tensor &query_neighbors_row_splits)

Constructor & Destructor Documentation

KnnSearchOpKernelCPU()

template<class T , class TIndex >

KnnSearchOpKernelCPU< T, TIndex >::KnnSearchOpKernelCPU ( OpKernelConstruction *  construction )

inlineexplicit

Member Function Documentation

Kernel()

template<class T , class TIndex >

void KnnSearchOpKernelCPU< T, TIndex >::Kernel ( tensorflow::OpKernelContext *  context, const tensorflow::Tensor &  points, const tensorflow::Tensor &  queries, const int  k, const tensorflow::Tensor &  points_row_splits, const tensorflow::Tensor &  queries_row_splits, tensorflow::Tensor &  query_neighbors_row_splits  )

inline

---

The documentation for this class was generated from the following file:

• /home/runner/work/Open3D/Open3D/cpp/open3d/ml/tensorflow/misc/KnnSearchOpKernel.cpp (252c867 (Sun May 5 05:39:34 2024 +0000))