Searched refs:maxNumDetections (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 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()
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3507 ANeuralNetworksOperandType maxNumDetections = getOpType(ANEURALNETWORKS_INT32); in detectionPostprocessingOpTest() local
3508 ANeuralNetworksOperandType numOfClassesPerDetection = maxNumDetections; in detectionPostprocessingOpTest()
3526 scaleScalar, isRegularNMS, maxNumDetections, numOfClassesPerDetection, in detectionPostprocessingOpTest()
|