Searched refs:scoreThreshold (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 388 float scoreThreshold, float iouThreshold, int32_t maxNumDetections, in hardNmsMultiClass() argument
398 if (score > scoreThreshold) { in hardNmsMultiClass()
422 uint32_t* softNmsSingleClass(float* scoresData, float scoreThreshold, int32_t maxNumDetections, in softNmsSingleClass() argument
441 if (scoresData[*i] < scoreThreshold) { in softNmsSingleClass()
452 float scoreThreshold, float nmsScoreThreshold, int32_t maxNumDetections, in softNmsMultiClass() argument
462 if (score > scoreThreshold) { in softNmsMultiClass()
488 float scoreThreshold, int32_t maxNumDetections, in boxWithNmsLimitFloat32Compute() argument
531 scoresBase, numClasses, batchSplitIn->at(b), scoreThreshold, nmsScoreThreshold, in boxWithNmsLimitFloat32Compute()
619 const Shape& batchesShape, float scoreThreshold, in boxWithNmsLimitFloat32() argument
633 scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, nmsScoreThreshold, in boxWithNmsLimitFloat32()
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3509 ANeuralNetworksOperandType scoreThreshold = scaleScalar; in detectionPostprocessingOpTest() local
3526 numOfDetections, scoreThreshold, iouThreshold, includeBackground}, in detectionPostprocessingOpTest()
|