KnnSearchOpKernelCPU< T > Class Template Reference

Inheritance diagram for KnnSearchOpKernelCPU< T >:

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 >

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

inlineexplicit

Member Function Documentation

Kernel()

template<class T >

void KnnSearchOpKernelCPU< T >::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 (313315d (Wed Dec 23 23:39:47 2020 -0800))