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

Method addConcatenation

include/NvInfer.h:7079–7082  ·  view source on GitHub ↗

\brief Add a concatenation layer to the network. \param inputs The input tensors to the layer. \param nbInputs The number of input tensors. \see IConcatenationLayer \return The new concatenation layer, or nullptr if it could not be created. \warning All tensors must have the same dimensions except along the concatenation axis.

Source from the content-addressed store, hash-verified

7077 //! \warning All tensors must have the same dimensions except along the concatenation axis.
7078 //!
7079 IConcatenationLayer* addConcatenation(ITensor* const* inputs, int32_t nbInputs) noexcept
7080 {
7081 return mImpl->addConcatenation(inputs, nbInputs);
7082 }
7083
7084 //!
7085 //! \brief Add a deconvolution layer to the network.

Callers 3

pyGraph.cppFile · 0.80
addLSTMLayersMethod · 0.80
parseConcatFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected