Inheritance diagram for nvinfer1::IGatherLayer:

Public Member Functions
virtual void	setGatherAxis (int axis)=0
	Set the non-batch dimension axis to gather on. The axis must be less than the number of non-batch dimensions in the data input. More...

virtual int	getGatherAxis () const =0
	Get the non-batch dimension axis to gather on. 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...

Member Function Documentation

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

pure virtual

Get the non-batch dimension axis to gather on.

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

pure virtual

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

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

Public Member Functions