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

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

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: