Searched refs:iouThreshold (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | GenerateProposals.cpp | 380 if (iou >= iouThreshold) { in hardNmsSingleClass()
391 float scoreThreshold, float iouThreshold, int32_t maxNumDetections, in hardNmsMultiClass() 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()
646 int32_t softNmsKernel, _Float16 iouThreshold, _Float16 sigma, in boxWithNmsLimitFloat16() argument
669 float iouThreshold, float sigma, float nmsScoreThreshold, in boxWithNmsLimitQuant() argument
691 float iouThreshold, float sigma, float nmsScoreThreshold, in boxWithNmsLimitQuant() argument
1039 scoresBase, iouThreshold, postNmsTopN, in generateProposalsNhwcFloat32Compute()
1076 heightStride, widthStride, preNmsTopN, postNmsTopN, iouThreshold, minSize, in generateProposalsFloat32Compute()
1496 scoreBase, numClasses, numAnchors, scoreThreshold, iouThreshold, in detectionPostprocessFloat32()
[all …]
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestValidateOperations.cpp | 3510 ANeuralNetworksOperandType iouThreshold = scaleScalar; in detectionPostprocessingOpTest() local
3526 numOfDetections, scoreThreshold, iouThreshold, includeBackground}, in detectionPostprocessingOpTest()
|