Open3D (C++ API)  0.18.0
Public Member Functions | Data Fields
SparseConvTransposeOpKernel< TIndex > Class Template Referenceabstract

#include <SparseConvTransposeOpKernel.h>

Inheritance diagram for SparseConvTransposeOpKernel< TIndex >:
SparseConvTransposeOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >

Public Member Functions

 SparseConvTransposeOpKernel (tensorflow::OpKernelConstruction *construction)
 
void Compute (tensorflow::OpKernelContext *context) override
 
virtual void Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &filters, 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 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

◆ SparseConvTransposeOpKernel()

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

Member Function Documentation

◆ Compute()

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

◆ Kernel()

template<class TIndex >
virtual void SparseConvTransposeOpKernel< TIndex >::Kernel ( tensorflow::OpKernelContext *  context,
const tensorflow::Tensor &  filters,
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 std::vector< int > &  filter_dims,
const bool  point_importances,
const bool  has_neighbors_importances,
tensorflow::Tensor &  out_features 
)
pure virtual

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

Field Documentation

◆ max_temp_mem_MB

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

◆ normalize

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