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

Method NormalizePluginCreator

plugin/normalizePlugin/normalizePlugin.cpp:285–296  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

283}
284
285NormalizePluginCreator::NormalizePluginCreator()
286{
287 mPluginAttributes.clear();
288 mPluginAttributes.emplace_back(PluginField("weights", nullptr, PluginFieldType::kFLOAT32, 1));
289 mPluginAttributes.emplace_back(PluginField("acrossSpatial", nullptr, PluginFieldType::kINT32, 1));
290 mPluginAttributes.emplace_back(PluginField("channelShared", nullptr, PluginFieldType::kINT32, 1));
291 mPluginAttributes.emplace_back(PluginField("nbWeights", nullptr, PluginFieldType::kINT32, 1));
292 mPluginAttributes.emplace_back(PluginField("eps", nullptr, PluginFieldType::kFLOAT32, 1));
293
294 mFC.nbFields = mPluginAttributes.size();
295 mFC.fields = mPluginAttributes.data();
296}
297
298char const* NormalizePluginCreator::getPluginName() const noexcept
299{

Callers

nothing calls this directly

Calls 4

PluginFieldClass · 0.85
clearMethod · 0.45
sizeMethod · 0.45
dataMethod · 0.45

Tested by

no test coverage detected