nvinfer1::IGatherLayer Class Reference

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IGatherLayer:

Public Member Functions
virtual void	setGatherAxis (int axis)=0
	Set the axis to gather on. The axis must be less than the number of dimensions in the data input. More...
virtual int	getGatherAxis () const =0
	Get the axis to gather on. More...
virtual void	setNbElementWiseDims (int k)=0
	Set the number of leading dimensions of indices tensor to be handled elementwise. k must be 0 if there is an implicit batch dimension. It can be 0 or 1 if there is not an implicit batch dimension. More...
virtual int	getNbElementWiseDims () const =0
	Get the number of leading dimensions of indices tensor to be handled elementwise. 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...
virtual void	setInput (int 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 (int index, DataType dataType)=0
	Set the output type of this layer. More...
virtual DataType	getOutputType (int index) const =0
	get the output type of this layer More...
virtual bool	outputTypeIsSet (int index) const =0
	whether the output type has been set for this layer More...
virtual void	resetOutputType (int index)=0
	reset the output type for this layer More...

Detailed Description

Warning

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

Member Function Documentation

virtual int nvinfer1::IGatherLayer::getGatherAxis ( ) const

pure virtual

Get the axis to gather on.

See also

setGatherAxis()

virtual int nvinfer1::IGatherLayer::getNbElementWiseDims ( ) const

pure virtual

Get the number of leading dimensions of indices tensor to be handled elementwise.

See also

setNbElementWiseDims()

virtual void nvinfer1::IGatherLayer::setGatherAxis ( int  axis )

pure virtual

Set the axis to gather on. The axis must be less than the number of dimensions in the data input.

See also

getGatherAxis()

virtual void nvinfer1::IGatherLayer::setNbElementWiseDims ( int  k )

pure virtual

Set the number of leading dimensions of indices tensor to be handled elementwise. k must be 0 if there is an implicit batch dimension. It can be 0 or 1 if there is not an implicit batch dimension.

See also

getNbElementWiseDims()

---

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

• NvInfer.h