Open3D (C++ API)  0.19.0
Loading...
Searching...
No Matches
Public Member Functions | Data Fields
SparseConvBackpropFilterOpKernel< TIndex > Class Template Referenceabstract

#include <SparseConvBackpropFilterOpKernel.h>

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

Public Member Functions

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

Data Fields

bool normalize
int max_temp_mem_MB

Constructor & Destructor Documentation

◆ SparseConvBackpropFilterOpKernel()

template<class TIndex>
SparseConvBackpropFilterOpKernel< TIndex >::SparseConvBackpropFilterOpKernel ( tensorflow::OpKernelConstruction * construction)
inlineexplicit

Member Function Documentation

◆ Compute()

template<class TIndex>
void SparseConvBackpropFilterOpKernel< TIndex >::Compute ( tensorflow::OpKernelContext * context)
inlineoverride

◆ Kernel()

template<class TIndex>
virtual void SparseConvBackpropFilterOpKernel< 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 tensorflow::Tensor & out_features_gradient,
const std::vector< int > & filter_dims,
const bool point_importances,
const bool has_neighbors_importances,
tensorflow::Tensor & filter_backprop )
pure virtual

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

Field Documentation

◆ max_temp_mem_MB

template<class TIndex>
int SparseConvBackpropFilterOpKernel< TIndex >::max_temp_mem_MB

◆ normalize

template<class TIndex>
bool SparseConvBackpropFilterOpKernel< TIndex >::normalize
The documentation for this class was generated from the following file:
Generated by doxygen 1.15.0