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

Method addLoopOutput

include/NvInfer.h:5710–5713  ·  view source on GitHub ↗

\brief Make an output for this loop, based on the given tensor. axis is the axis for concatenation (if using outputKind of kCONCATENATE or kREVERSE). If outputKind is kCONCATENATE or kREVERSE, a second input specifying the concatenation dimension must be added via method ILoopOutputLayer::setInput.

Source from the content-addressed store, hash-verified

5708 //! concatenation dimension must be added via method ILoopOutputLayer::setInput.
5709 //!
5710 ILoopOutputLayer* addLoopOutput(ITensor& tensor, LoopOutput outputKind, int32_t axis = 0) noexcept
5711 {
5712 return mImpl->addLoopOutput(tensor, outputKind, axis);
5713 }
5714
5715 //!
5716 //! \brief Set the name of the loop.

Callers 1

addLSTMCellMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected