SparseConvBackpropFilterOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex > Class Template Reference

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

Public Member Functions
	SparseConvBackpropFilterOpKernelCPU (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 tensorflow::Tensor &out_features_gradient, const std::vector< int > &filter_dims, const bool point_importances, const bool has_neighbors_importances, tensorflow::Tensor &filter_backprop)
Public Member Functions inherited from SparseConvBackpropFilterOpKernel< TIndex >
	SparseConvBackpropFilterOpKernel (tensorflow::OpKernelConstruction *construction)
void	Compute (tensorflow::OpKernelContext *context) override

Additional Inherited Members
Data Fields inherited from SparseConvBackpropFilterOpKernel< TIndex >
bool	normalize
int	max_temp_mem_MB

Constructor & Destructor Documentation

SparseConvBackpropFilterOpKernelCPU()

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

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

inlineexplicit

Member Function Documentation

Kernel()

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

void SparseConvBackpropFilterOpKernelCPU< 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 tensorflow::Tensor & out_features_gradient, const std::vector< int > & filter_dims, const bool point_importances, const bool has_neighbors_importances, tensorflow::Tensor & filter_backprop )

inlinevirtual

Implements SparseConvBackpropFilterOpKernel< TIndex >.

---

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

• cpp/open3d/ml/tensorflow/sparse_conv/SparseConvBackpropFilterOpKernel.cpp