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

Method setNbGroups

include/NvInfer.h:1193–1196  ·  view source on GitHub ↗

\brief Set the number of groups for a convolution. The input tensor channels are divided into \p nbGroups groups, and a convolution is executed for each group, using a filter per group. The results of the group convolutions are concatenated to form the output. \note When using groups in int8 mode, the size of the groups (i.e. the channel count divided by the group count) must be a multiple of 4

Source from the content-addressed store, hash-verified

1191 //! \see getNbGroups()
1192 //!
1193 void setNbGroups(int32_t nbGroups) noexcept
1194 {
1195 mImpl->setNbGroups(nbGroups);
1196 }
1197
1198 //!
1199 //! \brief Get the number of groups of the convolution.

Callers 4

parseConvolutionFunction · 0.45
parseDeconvolutionFunction · 0.45
setNbGroupsMethod · 0.45
setNbGroupsMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected