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

#include <SparseConvOpKernel.h>

Inheritance diagram for SparseConvOpKernel< TIndex >:
SparseConvOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >

Public Member Functions

 SparseConvOpKernel (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 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

◆ SparseConvOpKernel()

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

Member Function Documentation

◆ Compute()

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

◆ Kernel()

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

Field Documentation

◆ max_temp_mem_MB

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

◆ normalize

template<class TIndex >
bool SparseConvOpKernel< TIndex >::normalize

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