Home
last modified time | relevance | path

Searched refs:scoresBase (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DGenerateProposals.cpp522 float* scoresBase = scoresData; in boxWithNmsLimitFloat32Compute() local
541 return lhsClass == rhsClass ? scoresBase[lhs] > scoresBase[rhs] in boxWithNmsLimitFloat32Compute()
546 scoresBase += batchSplitIn->at(b) * numClasses; in boxWithNmsLimitFloat32Compute()
596 const float* scoresBase = scores.data(); in boxWithNmsLimitWriteOutput() local
613 scoresBase += batchSplitIn[b] * numClasses; in boxWithNmsLimitWriteOutput()
999 const float* scoresBase = scoresData; in generateProposalsNhwcFloat32Compute() local
1028 [&scoresBase](const uint32_t lhs, const uint32_t rhs) { in generateProposalsNhwcFloat32Compute()
1029 return scoresBase[lhs] > scoresBase[rhs]; in generateProposalsNhwcFloat32Compute()
1039 scoresBase, iouThreshold, postNmsTopN, in generateProposalsNhwcFloat32Compute()
1051 scoresOutData->push_back(scoresBase[i]); in generateProposalsNhwcFloat32Compute()
[all …]