Searched defs:numRois (Results 1 – 4 of 4) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| H A D | BoundingBox.cpp | 75 inline void fillRoiTensor(uint32_t numRois, T maxH, T maxW, RandomOperand* op) { in fillRoiTensor()
99 uint32_t numRois = op->inputs[1]->dimensions[0].getValue(); in roiFinalizer() local
187 RandomVariable numRois = RandomVariableType::FREE; in heatmapMaxKeypointConstructor() local
212 uint32_t numRois = op->inputs[0]->dimensions[0].getValue(); in heatmapMaxKeypointFinalizer() local
|
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | RoiPooling.cpp | 71 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc() local
242 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
|
| H A D | RoiAlign.cpp | 76 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
203 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignQuantNhwc() local
399 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
|
| H A D | GenerateProposals.cpp | 244 uint32_t numRois = getSizeOfDimension(roiShape, 0); 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()
508 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
774 uint32_t numRois = getSizeOfDimension(scoreShape, 0); in prepare() local
|