MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / addRNNv2

Method addRNNv2

include/NvInfer.h:7541–7545  ·  view source on GitHub ↗

\brief Add an \p layerCount deep RNN layer to the network with \p hiddenSize internal states that can take a batch with fixed or variable sequence lengths. \param input The input tensor to the layer (see below). \param layerCount The number of layers in the RNN. \param hiddenSize Size of the internal hidden state for each layer. \param maxSeqLen Maximum sequence length for the input. \param op Th

Source from the content-addressed store, hash-verified

7539 //! \return The new RNN layer, or nullptr if it could not be created.
7540 //!
7541 TRT_DEPRECATED IRNNv2Layer* addRNNv2(
7542 ITensor& input, int32_t layerCount, int32_t hiddenSize, int32_t maxSeqLen, RNNOperation op) noexcept
7543 {
7544 return mImpl->addRNNv2(input, layerCount, hiddenSize, maxSeqLen, op);
7545 }
7546
7547 //!
7548 //! \brief Add an identity layer.

Callers 1

addLSTMLayersMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected