Home
last modified time | relevance | path

Searched defs:numRois (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/
DBoundingBox.cpp67 inline void fillRoiTensor(uint32_t numRois, T maxH, T maxW, RandomOperand* op) { in fillRoiTensor()
89 uint32_t numRois = op->inputs[1]->dimensions[0].getValue(); in roiFinalizer() local
166 RandomVariable numRois = RandomVariableType::FREE; in heatmapMaxKeypointConstructor() local
187 uint32_t numRois = op->inputs[0]->dimensions[0].getValue(); in heatmapMaxKeypointFinalizer() local
/frameworks/ml/nn/common/operations/
DRoiPooling.cpp64 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiPoolingNhwc() local
214 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
DRoiAlign.cpp68 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
196 uint32_t numRois = getSizeOfDimension(roiShape, 0); in roiAlignNhwc() local
379 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
DGenerateProposals.cpp207 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
335 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass()
399 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass()
453 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
687 uint32_t numRois = getSizeOfDimension(scoreShape, 0); in prepare() local