Lines Matching refs:maxNumDetections
359 uint32_t* hardNmsSingleClass(const float* scoresData, float iouThreshold, int32_t maxNumDetections, in hardNmsSingleClass() argument
363 if (maxNumDetections < 0) { in hardNmsSingleClass()
364 maxNumDetections = selectLength; in hardNmsSingleClass()
366 while (selectStart < selectEnd && numDetections < maxNumDetections) { in hardNmsSingleClass()
388 float scoreThreshold, float iouThreshold, int32_t maxNumDetections, in hardNmsMultiClass() argument
414 if (maxNumDetections < 0 || select->size() <= maxNumDetections) { in hardNmsMultiClass()
417 select->resize(maxNumDetections); in hardNmsMultiClass()
422 uint32_t* softNmsSingleClass(float* scoresData, float scoreThreshold, int32_t maxNumDetections, in softNmsSingleClass() argument
426 if (maxNumDetections < 0) { in softNmsSingleClass()
427 maxNumDetections = selectLength; in softNmsSingleClass()
429 while (selectStart < selectEnd && numDetections < maxNumDetections) { in softNmsSingleClass()
452 float scoreThreshold, float nmsScoreThreshold, int32_t maxNumDetections, in softNmsMultiClass() argument
479 if (maxNumDetections < 0 || select->size() <= maxNumDetections) { in softNmsMultiClass()
482 select->resize(maxNumDetections); in softNmsMultiClass()
488 float scoreThreshold, int32_t maxNumDetections, in boxWithNmsLimitFloat32Compute() argument
532 maxNumDetections, maxNumDetections, in boxWithNmsLimitFloat32Compute()
620 int32_t maxNumDetections, int32_t softNmsKernel, float iouThreshold, in boxWithNmsLimitFloat32() argument
633 scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, nmsScoreThreshold, in boxWithNmsLimitFloat32()
642 _Float16 scoreThreshold, int32_t maxNumDetections, in boxWithNmsLimitFloat16() argument
656 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitFloat16()
665 float scoreThreshold, int32_t maxNumDetections, int32_t softNmsKernel, in boxWithNmsLimitQuant() argument
678 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitQuant()
687 float scoreThreshold, int32_t maxNumDetections, int32_t softNmsKernel, in boxWithNmsLimitQuant() argument
701 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitQuant()
1443 float scaleX, float scaleH, float scaleW, bool useRegularNms, int32_t maxNumDetections, in detectionPostprocessFloat32() argument
1486 maxNumDetections, maxNumDetectionsPerClass, in detectionPostprocessFloat32()
1513 maxScores.data(), iouThreshold, maxNumDetections, in detectionPostprocessFloat32()
1548 float scaleX, float scaleH, float scaleW, bool useRegularNms, int32_t maxNumDetections, in detectionPostprocessFloat16() argument
1564 maxNumDetections, maxClassesPerDetection, maxNumDetectionsPerClass, iouThreshold, in detectionPostprocessFloat16()
1624 uint32_t maxNumDetections = context->getInputValue<int32_t>(kMaxNumDetectionScalar); in prepare() local
1627 uint32_t numOutDetections = maxNumDetections; in prepare()
1653 NN_RET_CHECK_GT(maxNumDetections, 0); in prepare()