nvinfer1::IRecurrenceLayer Class Reference

Inheritance diagram for nvinfer1::IRecurrenceLayer:

Public Member Functions
void	setInput (int index, ITensor &tensor) _TENSORRT_OVERRIDE=0
	Append or replace an input of this layer with a specific tensor. More...
Public Member Functions inherited from nvinfer1::ILoopBoundaryLayer
virtual ILoop *	getLoop () const noexcept=0
	Return pointer to ILoop associated with this boundary layer.
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	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...

Member Function Documentation

void nvinfer1::IRecurrenceLayer::setInput ( int  index, ITensor &  tensor  )

pure virtual

Append or replace an input of this layer with a specific tensor.

Parameters

index	the index of the input to modify.
tensor	the new input tensor Sets the input tensor for the given index.

For a recurrence layer, the values 0 and 1 are valid. The indices are as follows:

Index | Description 0 | The initial value of the output tensor. The value must come from outside the loop. 1 | The next value of the output tensor. The value usually comes from inside the loop, and must have the same dimensions as input 0.

If this function is called with a value 1, then the function getNbInputs() changes from returning 1 to 2.

Implements nvinfer1::ILayer.

---

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

• NvInfer.h