Inheritance diagram for SparseConvOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >:

Public Member Functions
	SparseConvOpKernelCPU (OpKernelConstruction *construction)

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)

Public Member Functions inherited from SparseConvOpKernel< TIndex >
	SparseConvOpKernel (tensorflow::OpKernelConstruction *construction)

void	Compute (tensorflow::OpKernelContext *context) override

Additional Inherited Members
Data Fields inherited from SparseConvOpKernel< TIndex >
bool	normalize

int	max_temp_mem_MB

Constructor & Destructor Documentation

◆ SparseConvOpKernelCPU()

template<class TFeat , class TOut , class TIndex , class TKernelIndex >

SparseConvOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >::SparseConvOpKernelCPU ( OpKernelConstruction * construction )

inlineexplicit

template<class TFeat , class TOut , class TIndex , class TKernelIndex >

void SparseConvOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >::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
	)

inlinevirtual

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

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