Searched defs:numRois (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/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
|
| /frameworks/ml/nn/common/operations/ |
| D | RoiPooling.cpp | 70 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc() local
241 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
|
| D | RoiAlign.cpp | 74 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
201 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignQuantNhwc() local
396 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
|
| D | GenerateProposals.cpp | 243 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
387 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass()
451 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass()
505 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
769 uint32_t numRois = getSizeOfDimension(scoreShape, 0); in prepare() local
|