Searched defs:numClasses (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | GenerateProposals.cpp | 70 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / roiLength; in bboxTransformFloat32() local
212 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare() local
338 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass()
402 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass()
457 uint32_t numClasses = getSizeOfDimension(scoresShape, 1); in boxWithNmsLimitFloat32Compute() local
523 uint32_t numClasses = getSizeOfDimension(scoresShape, 1); in boxWithNmsLimitWriteOutput() local
681 uint32_t numClasses = getSizeOfDimension(scoreShape, 1); in prepare() local
1264 uint32_t numClasses = getSizeOfDimension(scoreShape, 2); in detectionPostprocessFloat32() local
1404 uint32_t numClasses = getSizeOfDimension(scoreShape, 2); in prepare() local
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3530 const int numClasses = 5; in detectionPostprocessingOpTest() local
|