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

Method addTripLimit

include/NvInfer.h:5685–5688  ·  view source on GitHub ↗

\brief Add a trip-count limiter, based on the given tensor. There may be at most one kCOUNT and one kWHILE limiter for a loop. When both trip limits exist, the loop exits when the count is reached or condition is falsified. It is an error to not add at least one trip limiter. For kCOUNT, the input tensor must be available before the loop starts. For kWHILE, the input tensor must be the output o

Source from the content-addressed store, hash-verified

5683 //! is the output of an IRecurrenceLayer for the same loop.
5684 //!
5685 ITripLimitLayer* addTripLimit(ITensor& tensor, TripLimit limit) noexcept
5686 {
5687 return mImpl->addTripLimit(tensor, limit);
5688 }
5689
5690 //!
5691 //! \brief Return layer that subscripts tensor by loop iteration.

Callers 1

addLSTMCellMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected