|
void | setAxis (int32_t axis) noexcept |
| Set axis to iterate over.
|
|
int32_t | getAxis () const noexcept |
| Get axis being iterated over.
|
|
void | setReverse (bool reverse) noexcept |
|
bool | getReverse () const noexcept |
| True if and only if reversing input.
|
|
ILoop * | getLoop () const noexcept |
| Return pointer to ILoop associated with this boundary layer.
|
|
LayerType | getType () const noexcept |
| Return the type of a layer. More...
|
|
void | setName (const char *name) noexcept |
| Set the name of a layer. More...
|
|
const char * | getName () const noexcept |
| Return the name of a layer. More...
|
|
int32_t | getNbInputs () const noexcept |
| Get the number of inputs of a layer.
|
|
ITensor * | getInput (int32_t index) const noexcept |
| Get the layer input corresponding to the given index. More...
|
|
int32_t | getNbOutputs () const noexcept |
| Get the number of outputs of a layer.
|
|
ITensor * | getOutput (int32_t index) const noexcept |
| Get the layer output corresponding to the given index. More...
|
|
void | setInput (int32_t index, ITensor &tensor) noexcept |
| Replace an input of this layer with a specific tensor. More...
|
|
void | setPrecision (DataType dataType) noexcept |
| Set the computational precision of this layer. More...
|
|
DataType | getPrecision () const noexcept |
| get the computational precision of this layer More...
|
|
bool | precisionIsSet () const noexcept |
| whether the computational precision has been set for this layer More...
|
|
void | resetPrecision () noexcept |
| reset the computational precision for this layer More...
|
|
void | setOutputType (int32_t index, DataType dataType) noexcept |
| Set the output type of this layer. More...
|
|
DataType | getOutputType (int32_t index) const noexcept |
| get the output type of this layer More...
|
|
bool | outputTypeIsSet (int32_t index) const noexcept |
| whether the output type has been set for this layer More...
|
|
void | resetOutputType (int32_t index) noexcept |
| reset the output type for this layer More...
|
|
void nvinfer1::IIteratorLayer::setReverse |
( |
bool |
reverse | ) |
|
|
inlinenoexcept |
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.