Open3D (C++ API)  0.18.0+252c867
Public Member Functions
SparseConvTransposeOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex > Class Template Reference
Inheritance diagram for SparseConvTransposeOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >:
SparseConvTransposeOpKernel< TIndex >

Public Member Functions

 SparseConvTransposeOpKernelCPU (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 std::vector< int > &filter_dims, const bool point_importances, const bool has_neighbors_importances, tensorflow::Tensor &out_features)
 
- Public Member Functions inherited from SparseConvTransposeOpKernel< TIndex >
 SparseConvTransposeOpKernel (tensorflow::OpKernelConstruction *construction)
 
void Compute (tensorflow::OpKernelContext *context) override
 

Additional Inherited Members

- Data Fields inherited from SparseConvTransposeOpKernel< TIndex >
bool normalize
 
int max_temp_mem_MB
 

Constructor & Destructor Documentation

◆ SparseConvTransposeOpKernelCPU()

template<class TFeat , class TOut , class TIndex , class TKernelIndex >
SparseConvTransposeOpKernelCPU< TFeat, TOut, TIndex, TKernelIndex >::SparseConvTransposeOpKernelCPU ( OpKernelConstruction *  construction)
inlineexplicit

Member Function Documentation

◆ Kernel()

template<class TFeat , class TOut , class TIndex , class TKernelIndex >
void SparseConvTransposeOpKernelCPU< 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 std::vector< int > &  filter_dims,
const bool  point_importances,
const bool  has_neighbors_importances,
tensorflow::Tensor &  out_features 
)
inlinevirtual

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