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

Method setNbGroups

include/NvInfer.h:2604–2607  ·  view source on GitHub ↗

\brief Set the number of groups for a deconvolution. The input tensor channels are divided into \p nbGroups groups, and a deconvolution is executed for each group, using a filter per group. The results of the group convolutions are concatenated to form the output. If executing this layer on DLA, nbGroups must be one \note When using groups in int8 mode, the size of the groups (i.e. the channel

Source from the content-addressed store, hash-verified

2602 //! \see getNbGroups()
2603 //!
2604 void setNbGroups(int32_t nbGroups) noexcept
2605 {
2606 mImpl->setNbGroups(nbGroups);
2607 }
2608
2609 //!
2610 //! \brief Get the number of groups for a deconvolution.

Callers

nothing calls this directly

Calls 1

setNbGroupsMethod · 0.45

Tested by

no test coverage detected