Searched defs:numClasses (Results 1 – 3 of 3) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A 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
|
| /aosp12/art/tools/dmtracedump/ |
| H A D | tracedump.cc | 231 int32_t numClasses; member
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestValidateOperations.cpp | 3492 const int numClasses = 5; in detectionPostprocessingOpTest() local
|