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

Method setKernelWeights

include/NvInfer.h:1217–1220  ·  view source on GitHub ↗

\brief Set the kernel weights for the convolution. The weights are specified as a contiguous array in \p GKCRS order, where \p G is the number of groups, \p K the number of output feature maps, \p C the number of input channels, and \p R and \p S are the height and width of the filter. \see getKernelWeights()

Source from the content-addressed store, hash-verified

1215 //! \see getKernelWeights()
1216 //!
1217 void setKernelWeights(Weights weights) noexcept
1218 {
1219 mImpl->setKernelWeights(weights);
1220 }
1221
1222 //!
1223 //! \brief Get the kernel weights of the convolution.

Callers 4

setSparseWeightsFunction · 0.45
parseDeconvolutionFunction · 0.45
setKernelWeightsFunction · 0.45
setKernelWeightsMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected