nvinfer1::IShapeLayer Class Reference
Layer type for getting shape of a tensor. More...
#include <NvInfer.h>
Inheritance diagram for nvinfer1::IShapeLayer:
Additional Inherited Members | |
 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 int32_t | getNbInputs () const =0 |
| Get the number of inputs of a layer. | |
| virtual ITensor * | getInput (int32_t index) const =0 |
| Get the layer input corresponding to the given index. More... | |
| virtual int32_t | getNbOutputs () const =0 |
| Get the number of outputs of a layer. | |
| virtual ITensor * | getOutput (int32_t index) const =0 |
| Get the layer output corresponding to the given index. More... | |
| virtual void | setInput (int32_t 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 (int32_t index, DataType dataType)=0 |
| Set the output type of this layer. More... | |
| virtual DataType | getOutputType (int32_t index) const =0 |
| get the output type of this layer More... | |
| virtual bool | outputTypeIsSet (int32_t index) const =0 |
| whether the output type has been set for this layer More... | |
| virtual void | resetOutputType (int32_t index)=0 |
| reset the output type for this layer More... | |
Detailed Description
Layer type for getting shape of a tensor.
This class sets the output to a one-dimensional tensor with the dimensions of the input tensor.
For example, if the input is a four-dimensional tensor (of any type) with dimensions [2,3,5,7], the output tensor is a one-dimensional Int32 tensor of length 4 containing the sequence 2, 3, 5, 7.
- Warning
- Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
The documentation for this class was generated from the following file:
