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

Method addParametricReLU

include/NvInfer.h:7756–7759  ·  view source on GitHub ↗

\brief Add a parametric ReLU layer to the network. \param input The input tensor to the layer. \param slope The slope tensor to the layer. This tensor should be unidirectionally broadcastable to the input tensor. \see IParametricReLULayer \warning Int32 tensors are not valid input tensors. \return The new parametric ReLU layer, or nullptr if it could not be created.

Source from the content-addressed store, hash-verified

7754 //! \return The new parametric ReLU layer, or nullptr if it could not be created.
7755 //!
7756 IParametricReLULayer* addParametricReLU(ITensor& input, ITensor& slope) noexcept
7757 {
7758 return mImpl->addParametricReLU(input, slope);
7759 }
7760
7761 //!
7762 //! \brief Add a multi-dimension convolution layer to the network.

Callers 1

parsePReLUFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected