Searched defs:numClasses (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | GenerateProposals.cpp | 68 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / roiLength; in bboxTransformFloat32() local
245 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare() local
390 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass()
454 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass()
509 uint32_t numClasses = getSizeOfDimension(scoresShape, 1); in boxWithNmsLimitFloat32Compute() local
574 uint32_t numClasses = getSizeOfDimension(scoresShape, 1); in boxWithNmsLimitWriteOutput() local
775 uint32_t numClasses = getSizeOfDimension(scoreShape, 1); in prepare() local
1462 uint32_t numClasses = getSizeOfDimension(scoreShape, 2); in detectionPostprocessFloat32() local
1635 uint32_t numClasses = getSizeOfDimension(scoreShape, 2); in prepare() local
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3492 const int numClasses = 5; in detectionPostprocessingOpTest() local
|