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 (int32_t k)=0
	Set the k value for the layer. More...
virtual int32_t	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 int32_t	getNbInputs () const =0
	Get the number of inputs of a layer.
virtual ITensor *	getInput (int32_t index) const =0
	Get the layer input corresponding to the given index. More...
virtual int32_t	getNbOutputs () const =0
	Get the number of outputs of a layer.
virtual ITensor *	getOutput (int32_t index) const =0
	Get the layer output corresponding to the given index. More...
virtual void	setInput (int32_t index, ITensor &tensor)=0
	Replace an input of this layer with a specific tensor. More...
virtual void	setPrecision (DataType dataType)=0
	Set the computational precision of this layer. More...
virtual DataType	getPrecision () const =0
	get the computational precision of this layer More...
virtual bool	precisionIsSet () const =0
	whether the computational precision has been set for this layer More...
virtual void	resetPrecision ()=0
	reset the computational precision for this layer More...
virtual void	setOutputType (int32_t index, DataType dataType)=0
	Set the output type of this layer. More...
virtual DataType	getOutputType (int32_t index) const =0
	get the output type of this layer More...
virtual bool	outputTypeIsSet (int32_t index) const =0
	whether the output type has been set for this layer More...
virtual void	resetOutputType (int32_t index)=0
	reset the output type for this layer More...

Detailed Description

Layer that represents a TopK reduction.

Warning

Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Member Function Documentation

getK()

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

pure virtual

Get the k value for the layer.

See also

setK()

getOperation()

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

pure virtual

Get the operation for the layer.

See also

setOperation(), TopKOperation

getReduceAxes()

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

pure virtual

Get the axes to reduce for the layer.

See also

setReduceAxes()

setK()

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

pure virtual

Set the k value for the layer.

Currently only values up to 3840 are supported.

See also

getK()

setOperation()

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

pure virtual

Set the operation for the layer.

See also

getOperation(), TopKOperation

setReduceAxes()

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