Searched refs:numClasses (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 67 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / roiLength; in bboxTransformFloat32() local
89 for (uint32_t i = 0; i < numClasses; i++) { in bboxTransformFloat32()
244 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare() local
246 NN_RET_CHECK_GT(numClasses, 0); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
262 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
387 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass() argument
393 for (uint32_t c = 1; c < numClasses; c++) { in hardNmsMultiClass()
396 const uint32_t index = b * numClasses + c; in hardNmsMultiClass()
451 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass() argument
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3493 const int numClasses = 5; in detectionPostprocessingOpTest() local
3496 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest()
|