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

Method setNbGroups

include/NvInfer.h:6797–6800  ·  view source on GitHub ↗

\brief Set the number of groups used to split the channels in the normalization calculation. The input tensor channels are divided into \p nbGroups groups, and normalization is performed per group. The channel dimension is considered to be the second dimension in a [N, C, H, W, ...] formatted tensor. The default \p nbGroups is 1. \warning It is an error to set \p nbGroups to a value that does n

Source from the content-addressed store, hash-verified

6795 //! \param nbGroups The number of groups to split the channels into for the normalization calculation.
6796 //!
6797 void setNbGroups(int32_t nbGroups) noexcept
6798 {
6799 return mImpl->setNbGroups(nbGroups);
6800 }
6801
6802 //! \brief Get the number of groups used to split the channels for the normalization calculation.
6803 //!

Callers

nothing calls this directly

Calls 1

setNbGroupsMethod · 0.45

Tested by

no test coverage detected