Searched refs:numAnchors (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 965 uint32_t numAnchors = getSizeOfDimension(scoresShape, 3); in generateProposalsNhwcFloat32Compute() local
968 uint32_t batchSize = height * width * numAnchors; in generateProposalsNhwcFloat32Compute()
983 for (uint32_t a = 0; a < numAnchors; a++, roiBase += kRoiDim, anchorsBase += kRoiDim) { in generateProposalsNhwcFloat32Compute()
1299 uint32_t numAnchors = getSizeOfDimension(scoreShape, useNchw ? 1 : 3); in prepare() local
1304 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, useNchw ? 1 : 3), numAnchors * kRoiDim); in prepare()
1307 NN_RET_CHECK_EQ(getSizeOfDimension(anchorsShape, 0), numAnchors); in prepare()
1450 uint32_t numAnchors = getSizeOfDimension(scoreShape, 1); in detectionPostprocessFloat32() local
1465 std::vector<float> roiBuffer(numAnchors * kRoiDim); in detectionPostprocessFloat32()
1466 std::vector<float> scoreBuffer(numAnchors); in detectionPostprocessFloat32()
1469 for (uint32_t a = 0; a < numAnchors; a++) { in detectionPostprocessFloat32()
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3492 const int numAnchors = 10; in detectionPostprocessingOpTest() local
3496 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest()
3498 uint32_t deltasDims[3] = {numBatches, numAnchors, lengthBoxEncoding}; in detectionPostprocessingOpTest()
3500 uint32_t anchorsDims[2] = {numAnchors, 4}; in detectionPostprocessingOpTest()
|