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

Method addSoftMax

include/NvInfer.h:7062–7065  ·  view source on GitHub ↗

\brief Add a SoftMax layer to the network. \see ISoftMaxLayer \warning Int32 tensors are not valid input tensors. \return The new SoftMax layer, or nullptr if it could not be created.

Source from the content-addressed store, hash-verified

7060 //! \return The new SoftMax layer, or nullptr if it could not be created.
7061 //!
7062 ISoftMaxLayer* addSoftMax(ITensor& input) noexcept
7063 {
7064 return mImpl->addSoftMax(input);
7065 }
7066
7067 //!
7068 //! \brief Add a concatenation layer to the network.

Callers 2

buildPredictionEngineMethod · 0.80
parseSoftMaxFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected