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

Method addUnary

include/NvInfer.h:7152–7155  ·  view source on GitHub ↗

\brief Add a unary layer to the network. \param input The input tensor to the layer. \param operation The operation to apply. \see IUnaryLayer Generally the input must have a floating-point type (or kINT8 as a quantized float), except for the following operations: kSIGN accepts a floating-point or Int32 tensor. kNOT requires a Bool tensor. The input is a shape tensor if the output is a shape t

Source from the content-addressed store, hash-verified

7150 //! \return The new unary layer, or nullptr if it could not be created
7151 //!
7152 IUnaryLayer* addUnary(ITensor& input, UnaryOperation operation) noexcept
7153 {
7154 return mImpl->addUnary(input, operation);
7155 }
7156
7157 //! \brief Add a padding layer to the network.
7158 //!

Callers 2

parseAbsValFunction · 0.80
parseReductionFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected