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

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

Public Member Functions
	SparseConvTransposeBackpropFilterOpKernelCPU (OpKernelConstruction *construction)
void	Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &filter, const tensorflow::Tensor &out_importance, const tensorflow::Tensor &inp_features, const tensorflow::Tensor &inp_neighbors_importance_sum, const tensorflow::Tensor &inp_neighbors_row_splits, 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 SparseConvTransposeBackpropFilterOpKernel< TIndex >
	SparseConvTransposeBackpropFilterOpKernel (tensorflow::OpKernelConstruction *construction)
void	Compute (tensorflow::OpKernelContext *context) override

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

Constructor & Destructor Documentation

SparseConvTransposeBackpropFilterOpKernelCPU()

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

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

inlineexplicit

Member Function Documentation

Kernel()

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

void SparseConvTransposeBackpropFilterOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >::Kernel ( tensorflow::OpKernelContext *  context, const tensorflow::Tensor &  filter, const tensorflow::Tensor &  out_importance, const tensorflow::Tensor &  inp_features, const tensorflow::Tensor &  inp_neighbors_importance_sum, const tensorflow::Tensor &  inp_neighbors_row_splits, 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 SparseConvTransposeBackpropFilterOpKernel< TIndex >.

---

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

• /home/runner/work/Open3D/Open3D/cpp/open3d/ml/tensorflow/sparse_conv/SparseConvTransposeBackpropFilterOpKernel.cpp (b9e049c (Sun Dec 31 11:36:26 2023 -0800))