| 56 | std::vector<PluginField> BatchedNMSBasePluginCreator::mPluginAttributes; |
| 57 | |
| 58 | static inline pluginStatus_t checkParams(NMSParameters const& param) |
| 59 | { |
| 60 | // NMS plugin supports maximum thread blocksize of 512 and upto 8 blocks at once. |
| 61 | constexpr int32_t maxTopK{512 * 8}; |
| 62 | if (param.topK > maxTopK) |
| 63 | { |
| 64 | plugin::gLogError << "Invalid parameter: NMS topK (" << param.topK << ") exceeds limit (" << maxTopK << ")" |
| 65 | << std::endl; |
| 66 | return STATUS_BAD_PARAM; |
| 67 | } |
| 68 | |
| 69 | return STATUS_SUCCESS; |
| 70 | } |
| 71 | |
| 72 | BatchedNMSPlugin::BatchedNMSPlugin(NMSParameters params) |
| 73 | : param(params) |
no outgoing calls
no test coverage detected