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:

• cpp/open3d/ml/tensorflow/misc/KnnSearchOpKernel.cpp