| 234 | } |
| 235 | |
| 236 | size_t BatchedNMSPlugin::getWorkspaceSize(int32_t maxBatchSize) const noexcept |
| 237 | { |
| 238 | return detectionInferenceWorkspaceSize(param.shareLocation, maxBatchSize, mBoxesSize, mScoresSize, param.numClasses, |
| 239 | mNumPriors, param.topK, mPrecision, mPrecision); |
| 240 | } |
| 241 | |
| 242 | size_t BatchedNMSDynamicPlugin::getWorkspaceSize( |
| 243 | PluginTensorDesc const* inputs, int32_t nbInputs, PluginTensorDesc const* outputs, int32_t nbOutputs) const noexcept |
nothing calls this directly
no test coverage detected