TensorRT
8.0.3
|
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 |
Copyright © 2024 NVIDIA Corporation
Privacy Policy |
Manage My Privacy |
Do Not Sell or Share My Data |
Terms of Service |
Accessibility |
Corporate Policies |
Product Security |
Contact