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

Method addElementWise

include/NvInfer.h:7130–7133  ·  view source on GitHub ↗

\brief Add an elementwise layer to the network. \param input1 The first input tensor to the layer. \param input2 The second input tensor to the layer. \param op The binary operation that the layer applies. The input tensors must have the same rank and compatible type. Two types are compatible if they are the same type or are both in the set {kFLOAT, kHALF}. For each dimension, their lengths must

Source from the content-addressed store, hash-verified

7128 //! \return The new elementwise layer, or nullptr if it could not be created.
7129 //!
7130 IElementWiseLayer* addElementWise(ITensor& input1, ITensor& input2, ElementWiseOperation op) noexcept
7131 {
7132 return mImpl->addElementWise(input1, input2, op);
7133 }
7134
7135 //!
7136 //! \brief Add a unary layer to the network.

Callers 4

addLSTMCellMethod · 0.80
constructNetworkMethod · 0.80
parseEltwiseFunction · 0.80
parseReductionFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected