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

Method addConvolutionNd

include/NvInfer.h:7778–7782  ·  view source on GitHub ↗

\brief Add a multi-dimension convolution layer to the network. \param input The input tensor to the convolution. \param nbOutputMaps The number of output feature maps for the convolution. \param kernelSize The multi-dimensions of the convolution kernel. \param kernelWeights The kernel weights for the convolution. \param biasWeights The bias weights for the convolution. Weights{} represents no bia

Source from the content-addressed store, hash-verified

7776 //! \return The new convolution layer, or nullptr if it could not be created.
7777 //!
7778 IConvolutionLayer* addConvolutionNd(
7779 ITensor& input, int32_t nbOutputMaps, Dims kernelSize, Weights kernelWeights, Weights biasWeights) noexcept
7780 {
7781 return mImpl->addConvolutionNd(input, nbOutputMaps, kernelSize, kernelWeights, biasWeights);
7782 }
7783
7784 //!
7785 //! \brief Add a multi-dimension pooling layer to the network.

Callers 1

pyGraph.cppFile · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected