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

Method addRecurrence

include/NvInfer.h:5664–5667  ·  view source on GitHub ↗

\brief Create a recurrence layer for this loop with initialValue as its first input. IRecurrenceLayer requires exactly two inputs. The 2nd input must be added, via method IRecurrenceLayer::setInput(1,...) before an Engine can be built.

Source from the content-addressed store, hash-verified

5662 //! IRecurrenceLayer::setInput(1,...) before an Engine can be built.
5663 //!
5664 IRecurrenceLayer* addRecurrence(ITensor& initialValue) noexcept
5665 {
5666 return mImpl->addRecurrence(initialValue);
5667 }
5668
5669 //!
5670 //! \brief Add a trip-count limiter, based on the given tensor.

Callers 1

addLSTMCellMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected