nvinfer1::ITopKLayer Class Reference

Layer that represents a TopK reduction. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::ITopKLayer:

Public Member Functions
virtual void	setOperation (TopKOperation op)=0
	Set the operation for the layer. More...
virtual TopKOperation	getOperation () const =0
	Get the operation for the layer. More...
virtual void	setK (int k)=0
	Set the k value for the layer. More...
virtual int	getK () const =0
	Get the k value for the layer. More...
virtual void	setReduceAxes (uint32_t reduceAxes)=0
	Set which axes to reduce for the layer. More...
virtual uint32_t	getReduceAxes () const =0
	Get the axes to reduce for the layer. More...
Public Member Functions inherited from nvinfer1::ILayer
virtual LayerType	getType () const =0
	Return the type of a layer. More...
virtual void	setName (const char *name)=0
	Set the name of a layer. More...
virtual const char *	getName () const =0
	Return the name of a layer. More...
virtual int	getNbInputs () const =0
	Get the number of inputs of a layer.
virtual ITensor *	getInput (int index) const =0
	Get the layer input corresponding to the given index. More...
virtual int	getNbOutputs () const =0
	Get the number of outputs of a layer.
virtual ITensor *	getOutput (int index) const =0
	Get the layer output corresponding to the given index. More...

Detailed Description

Layer that represents a TopK reduction.

Member Function Documentation

virtual int nvinfer1::ITopKLayer::getK ( ) const

pure virtual

Get the k value for the layer.

See Also

setK()

virtual TopKOperation nvinfer1::ITopKLayer::getOperation ( ) const

pure virtual

Get the operation for the layer.

See Also

setOperation(), TopKOperation

virtual uint32_t nvinfer1::ITopKLayer::getReduceAxes ( ) const

pure virtual

Get the axes to reduce for the layer.

See Also

setReduceAxes()

virtual void nvinfer1::ITopKLayer::setK ( int  k )

pure virtual

Set the k value for the layer.

Currently only values up to 25 are supported.

See Also

getK()

virtual void nvinfer1::ITopKLayer::setOperation ( TopKOperation  op )

pure virtual

Set the operation for the layer.

See Also

getOperation(), TopKOperation

virtual void nvinfer1::ITopKLayer::setReduceAxes ( uint32_t  reduceAxes )

pure virtual

Set which axes to reduce for the layer.

See Also

getReduceAxes()

---

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

• NvInfer.h