SparseConvOpKernel< TIndex > Class Template Reference

#include <SparseConvOpKernel.h>

Inheritance diagram for SparseConvOpKernel< TIndex >:

Public Member Functions
	SparseConvOpKernel (tensorflow::OpKernelConstruction *construction)
void	Compute (tensorflow::OpKernelContext *context) override
virtual void	Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &filters, const tensorflow::Tensor &inp_features, const tensorflow::Tensor &inp_importance, const tensorflow::Tensor &neighbors_index, const tensorflow::Tensor &neighbors_kernel_index, const tensorflow::Tensor &neighbors_importance, const tensorflow::Tensor &neighbors_row_splits, const std::vector< int > &filter_dims, const bool point_importances, const bool has_neighbors_importances, tensorflow::Tensor &out_features)=0

Data Fields
bool	normalize
int	max_temp_mem_MB

Constructor & Destructor Documentation

SparseConvOpKernel()

template<class TIndex >

SparseConvOpKernel< TIndex >::SparseConvOpKernel ( tensorflow::OpKernelConstruction *  construction )

inlineexplicit

Member Function Documentation

Compute()

template<class TIndex >

void SparseConvOpKernel< TIndex >::Compute ( tensorflow::OpKernelContext *  context )

inlineoverride

Kernel()

template<class TIndex >

virtual void SparseConvOpKernel< TIndex >::Kernel ( tensorflow::OpKernelContext *  context, const tensorflow::Tensor &  filters, const tensorflow::Tensor &  inp_features, const tensorflow::Tensor &  inp_importance, const tensorflow::Tensor &  neighbors_index, const tensorflow::Tensor &  neighbors_kernel_index, const tensorflow::Tensor &  neighbors_importance, const tensorflow::Tensor &  neighbors_row_splits, const std::vector< int > &  filter_dims, const bool  point_importances, const bool  has_neighbors_importances, tensorflow::Tensor &  out_features  )

pure virtual

Implemented in SparseConvOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >.

Field Documentation

max_temp_mem_MB

template<class TIndex >

int SparseConvOpKernel< TIndex >::max_temp_mem_MB

normalize

template<class TIndex >

bool SparseConvOpKernel< TIndex >::normalize

---

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

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