Searched defs:numRois (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/ |
| 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
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RoiAlign.cpp | 64 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
191 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignQuantNhwc() local
344 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
|
| D | RoiPooling.cpp | 58 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc() local
190 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
|
| D | GenerateProposals.cpp | 212 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
339 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass()
403 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass()
457 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
681 uint32_t numRois = getSizeOfDimension(scoreShape, 0); in prepare() local
|