|
TensorRT
7.0.0.11
|
|
TensorRT
7.0.0.11
|
Public Member Functions | |
| virtual void | setAxis (int axis) noexcept=0 |
| Set axis to iterate over. | |
| virtual int | getAxis () const noexcept=0 |
| Get axis being iterated over. | |
| virtual void | setReverse (bool reverse) noexcept=0 |
| virtual bool | getReverse () const noexcept=0 |
| True if and only if reversing input. | |
 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 | 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... | |
|
pure virtualnoexcept |
For reverse=false, the layer is equivalent to addGather(tensor, I, 0) where I is a scalar tensor containing the loop iteration number. For reverse=true, the layer is equivalent to addGather(tensor, M-1-I, 0) where M is the trip count computed from TripLimits of kind kCOUNT. The default is reverse=false.
1.8.11 
