Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVIDIA/TensorRT
/ getWeightsSize
Function
getWeightsSize
plugin/common/bertCommon.h:147–150 ·
view source on GitHub ↗
Source
from the content-addressed store, hash-verified
145
}
146
147
inline int64_t getWeightsSize(nvinfer1::Weights const& w, nvinfer1::DataType type)
148
{
149
return w.count * getElementSize(type);
150
}
151
152
inline int64_t volume(nvinfer1::Dims const& d)
153
{
Callers
8
FCPluginDynamic
Method · 0.85
EmbLayerNormPluginDynamic
Method · 0.85
EmbLayerNormVarSeqlenPluginBase
Method · 0.85
configurePlugin
Method · 0.85
SkipLayerNormPluginDynamic
Method · 0.85
SkipLayerNormVarSeqlenPlugin
Method · 0.85
GeluPluginDynamic
Method · 0.85
convertAndCopy
Method · 0.85
Calls
1
getElementSize
Function · 0.70
Tested by
no test coverage detected