|
TensorRT
|
Layer that represents a constant value. More...
#include <NvInfer.h>
Public Member Functions | |
| virtual void | setWeights (Weights weights)=0 |
| Set the weights for the layer. More... | |
| virtual Weights | getWeights () const =0 |
| Get the weights for the layer. More... | |
| virtual void | setDimensions (Dims dimensions)=0 |
| Set the dimensions for the layer. More... | |
| virtual Dims | getDimensions () const =0 |
| Get the dimensions for the layer. 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... | |
Layer that represents a constant value.
|
pure virtual |
|
pure virtual |
Get the weights for the layer.
|
pure virtual |
Set the dimensions for the layer.
| dimensions | The dimensions of the layer |
|
pure virtual |
Set the weights for the layer.