Inheritance diagram for InvertNeighborsListOpKernelCPU< TIndex, TAttr >:

Public Member Functions
	InvertNeighborsListOpKernelCPU (OpKernelConstruction *construction)

void	Kernel (tensorflow::OpKernelContext *context, const tensorflow::Tensor &inp_neighbors_index, const tensorflow::Tensor &inp_neighbors_row_splits, const tensorflow::Tensor &inp_neighbors_attributes, const int num_attributes, tensorflow::Tensor &neighbors_index, tensorflow::Tensor &neighbors_row_splits, tensorflow::Tensor &neighbors_attributes)

Constructor & Destructor Documentation

◆ InvertNeighborsListOpKernelCPU()

template<class TIndex , class TAttr >

InvertNeighborsListOpKernelCPU< TIndex, TAttr >::InvertNeighborsListOpKernelCPU ( OpKernelConstruction * construction )

inlineexplicit

template<class TIndex , class TAttr >

void InvertNeighborsListOpKernelCPU< TIndex, TAttr >::Kernel	(	tensorflow::OpKernelContext *	context,
		const tensorflow::Tensor &	inp_neighbors_index,
		const tensorflow::Tensor &	inp_neighbors_row_splits,
		const tensorflow::Tensor &	inp_neighbors_attributes,
		const int	num_attributes,
		tensorflow::Tensor &	neighbors_index,
		tensorflow::Tensor &	neighbors_row_splits,
		tensorflow::Tensor &	neighbors_attributes
	)

inline

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