nvinfer1::apiv::VRNNv2Layer Class Referenceabstract
Inheritance diagram for nvinfer1::apiv::VRNNv2Layer:
Public Member Functions | |
| virtual int32_t | getLayerCount () const noexcept=0 |
| virtual int32_t | getHiddenSize () const noexcept=0 |
| virtual int32_t | getMaxSeqLength () const noexcept=0 |
| virtual int32_t | getDataLength () const noexcept=0 |
| virtual void | setSequenceLengths (ITensor &seqLengths) noexcept=0 |
| virtual ITensor * | getSequenceLengths () const noexcept=0 |
| virtual void | setOperation (RNNOperation op) noexcept=0 |
| virtual RNNOperation | getOperation () const noexcept=0 |
| virtual void | setInputMode (RNNInputMode op) noexcept=0 |
| virtual RNNInputMode | getInputMode () const noexcept=0 |
| virtual void | setDirection (RNNDirection op) noexcept=0 |
| virtual RNNDirection | getDirection () const noexcept=0 |
| virtual void | setWeightsForGate (int32_t layerIndex, RNNGateType gate, bool isW, Weights weights) noexcept=0 |
| virtual Weights | getWeightsForGate (int32_t layerIndex, RNNGateType gate, bool isW) const noexcept=0 |
| virtual void | setBiasForGate (int32_t layerIndex, RNNGateType gate, bool isW, Weights bias) noexcept=0 |
| virtual Weights | getBiasForGate (int32_t layerIndex, RNNGateType gate, bool isW) const noexcept=0 |
| virtual void | setHiddenState (ITensor &hidden) noexcept=0 |
| virtual ITensor * | getHiddenState () const noexcept=0 |
| virtual void | setCellState (ITensor &cell) noexcept=0 |
| virtual ITensor * | getCellState () const noexcept=0 |
The documentation for this class was generated from the following file:
