Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DGenerateProposals.cpp972 uint32_t numAnchors = getSizeOfDimension(scoresShape, 3); in generateProposalsNhwcFloat32Compute() local
975 uint32_t batchSize = height * width * numAnchors; in generateProposalsNhwcFloat32Compute()
1461 uint32_t numAnchors = getSizeOfDimension(scoreShape, 1); in detectionPostprocessFloat32() local
1476 std::vector<float> roiBuffer(numAnchors * kRoiDim); in detectionPostprocessFloat32()
1477 std::vector<float> scoreBuffer(numAnchors); in detectionPostprocessFloat32()
1480 for (uint32_t a = 0; a < numAnchors; a++) { in detectionPostprocessFloat32()
1512 std::vector<float> maxScores(numAnchors); in detectionPostprocessFloat32()
1513 for (uint32_t a = 0; a < numAnchors; a++) { in detectionPostprocessFloat32()
1518 for (uint32_t a = 0; a < numAnchors; a++) { in detectionPostprocessFloat32()
1548 scoreBase += numAnchors * numClasses; in detectionPostprocessFloat32()
[all …]
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestValidateOperations.cpp3491 const int numAnchors = 10; in detectionPostprocessingOpTest() local
3495 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest()
3497 uint32_t deltasDims[3] = {numBatches, numAnchors, lengthBoxEncoding}; in detectionPostprocessingOpTest()
3499 uint32_t anchorsDims[2] = {numAnchors, 4}; in detectionPostprocessingOpTest()