Open3D (C++ API)  0.19.0
Loading...
Searching...
No Matches
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.15.0