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

Method ProposalBasePluginCreator

plugin/proposalPlugin/proposalPlugin.cpp:637–651  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

635void ProposalPlugin::detachFromContext() noexcept {}
636
637ProposalBasePluginCreator::ProposalBasePluginCreator() noexcept
638{
639 mPluginAttributes.clear();
640 mPluginAttributes.emplace_back(PluginField("input_height", nullptr, PluginFieldType::kINT32, 1));
641 mPluginAttributes.emplace_back(PluginField("input_width", nullptr, PluginFieldType::kINT32, 1));
642 mPluginAttributes.emplace_back(PluginField("rpn_stride", nullptr, PluginFieldType::kINT32, 1));
643 mPluginAttributes.emplace_back(PluginField("roi_min_size", nullptr, PluginFieldType::kFLOAT32, 1));
644 mPluginAttributes.emplace_back(PluginField("nms_iou_threshold", nullptr, PluginFieldType::kFLOAT32, 1));
645 mPluginAttributes.emplace_back(PluginField("pre_nms_top_n", nullptr, PluginFieldType::kINT32, 1));
646 mPluginAttributes.emplace_back(PluginField("post_nms_top_n", nullptr, PluginFieldType::kINT32, 1));
647 mPluginAttributes.emplace_back(PluginField("anchor_sizes", nullptr, PluginFieldType::kFLOAT32, 1));
648 mPluginAttributes.emplace_back(PluginField("anchor_ratios", nullptr, PluginFieldType::kFLOAT32, 1));
649 mFC.nbFields = mPluginAttributes.size();
650 mFC.fields = mPluginAttributes.data();
651}
652
653ProposalPluginCreator::ProposalPluginCreator() noexcept
654{

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