MCPcopy 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
147inline int64_t getWeightsSize(nvinfer1::Weights const& w, nvinfer1::DataType type)
148{
149 return w.count * getElementSize(type);
150}
151
152inline int64_t volume(nvinfer1::Dims const& d)
153{

Callers 8

FCPluginDynamicMethod · 0.85
configurePluginMethod · 0.85
GeluPluginDynamicMethod · 0.85
convertAndCopyMethod · 0.85

Calls 1

getElementSizeFunction · 0.70

Tested by

no test coverage detected