Lines Matching refs:iouThreshold

362 uint32_t* hardNmsSingleClass(const float* scoresData, float iouThreshold, int32_t maxNumDetections,  in hardNmsSingleClass()  argument
380 if (iou >= iouThreshold) { in hardNmsSingleClass()
391 float scoreThreshold, float iouThreshold, int32_t maxNumDetections, in hardNmsMultiClass() argument
407 uint32_t* selectEnd = hardNmsSingleClass(scoresData, iouThreshold, maxNumDetectionsPerClass, in hardNmsMultiClass()
492 int32_t softNmsKernel, float iouThreshold, float sigma, in boxWithNmsLimitFloat32Compute() argument
498 kernel = [&iouThreshold](float iou) { return iou < iouThreshold ? 1.0f : 0.0f; }; in boxWithNmsLimitFloat32Compute()
500 kernel = [&iouThreshold](float iou) { return iou < iouThreshold ? 1.0f : 1.0f - iou; }; in boxWithNmsLimitFloat32Compute()
623 int32_t maxNumDetections, int32_t softNmsKernel, float iouThreshold, in boxWithNmsLimitFloat32() argument
636 scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, nmsScoreThreshold, in boxWithNmsLimitFloat32()
646 int32_t softNmsKernel, _Float16 iouThreshold, _Float16 sigma, in boxWithNmsLimitFloat16() argument
659 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitFloat16()
669 float iouThreshold, float sigma, float nmsScoreThreshold, in boxWithNmsLimitQuant() argument
681 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitQuant()
691 float iouThreshold, float sigma, float nmsScoreThreshold, in boxWithNmsLimitQuant() argument
704 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitQuant()
964 int32_t postNmsTopN, float iouThreshold, float minSize, in generateProposalsNhwcFloat32Compute() argument
1039 scoresBase, iouThreshold, postNmsTopN, in generateProposalsNhwcFloat32Compute()
1066 int32_t postNmsTopN, float iouThreshold, float minSize, in generateProposalsFloat32Compute() argument
1076 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, in generateProposalsFloat32Compute()
1085 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat32() argument
1092 iouThreshold, minSize, useNchw, &scoresOut_float32, &roiOut_float32, &batchesOut)); in generateProposalsFloat32()
1128 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsFloat16() argument
1143 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsFloat16()
1177 int32_t postNmsTopN, float iouThreshold, float minSize, bool useNchw, in generateProposalsQuant() argument
1195 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, useNchw, in generateProposalsQuant()
1455 int32_t maxClassesPerDetection, int32_t maxNumDetectionsPerClass, float iouThreshold, in detectionPostprocessFloat32() argument
1496 scoreBase, numClasses, numAnchors, scoreThreshold, iouThreshold, in detectionPostprocessFloat32()
1524 maxScores.data(), iouThreshold, maxNumDetections, in detectionPostprocessFloat32()
1560 int32_t maxClassesPerDetection, int32_t maxNumDetectionsPerClass, float iouThreshold, in detectionPostprocessFloat16() argument
1575 maxNumDetections, maxClassesPerDetection, maxNumDetectionsPerClass, iouThreshold, in detectionPostprocessFloat16()