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

Method addShuffle

include/NvInfer.h:7183–7186  ·  view source on GitHub ↗

\brief Add a shuffle layer to the network. \param input The input tensor to the layer. \see IShuffleLayer \return The new shuffle layer, or nullptr if it could not be created.

Source from the content-addressed store, hash-verified

7181 //! \return The new shuffle layer, or nullptr if it could not be created.
7182 //!
7183 IShuffleLayer* addShuffle(ITensor& input) noexcept
7184 {
7185 return mImpl->addShuffle(input);
7186 }
7187
7188 //!
7189 //! \brief Add a OneHot layer to the network.

Callers 3

addReshapeMethod · 0.80
parseReshapeFunction · 0.80
parsePermuteFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected