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

Method addActivation

include/NvInfer.h:6984–6987  ·  view source on GitHub ↗

\brief Add an activation layer to the network. \param input The input tensor to the layer. \param type The type of activation function to apply. Note that the setAlpha() and setBeta() methods must be used on the output for activations that require these parameters. \see IActivationLayer ActivationType \warning Int32 tensors are not valid input tensors. \return The new activation layer, or null

Source from the content-addressed store, hash-verified

6982 //! \return The new activation layer, or nullptr if it could not be created.
6983 //!
6984 IActivationLayer* addActivation(ITensor& input, ActivationType type) noexcept
6985 {
6986 return mImpl->addActivation(input, type);
6987 }
6988
6989 //!
6990 //! \brief Add a pooling layer to the network.

Callers 7

addLSTMCellMethod · 0.80
parseReLUFunction · 0.80
parseSigmoidFunction · 0.80
parseClipFunction · 0.80
parseELUFunction · 0.80
parseBNLLFunction · 0.80
parseTanHFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected