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: