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

Method GridAnchorBasePluginCreator

plugin/gridAnchorPlugin/gridAnchorPlugin.cpp:369–381  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

367}
368
369GridAnchorBasePluginCreator::GridAnchorBasePluginCreator()
370{
371 mPluginAttributes.clear();
372 mPluginAttributes.emplace_back(PluginField("minSize", nullptr, PluginFieldType::kFLOAT32, 1));
373 mPluginAttributes.emplace_back(PluginField("maxSize", nullptr, PluginFieldType::kFLOAT32, 1));
374 mPluginAttributes.emplace_back(PluginField("aspectRatios", nullptr, PluginFieldType::kFLOAT32, 1));
375 mPluginAttributes.emplace_back(PluginField("featureMapShapes", nullptr, PluginFieldType::kINT32, 1));
376 mPluginAttributes.emplace_back(PluginField("variance", nullptr, PluginFieldType::kFLOAT32, 4));
377 mPluginAttributes.emplace_back(PluginField("numLayers", nullptr, PluginFieldType::kINT32, 1));
378
379 mFC.nbFields = mPluginAttributes.size();
380 mFC.fields = mPluginAttributes.data();
381}
382
383char const* GridAnchorBasePluginCreator::getPluginName() const noexcept
384{

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