Lines Matching refs:numRois
212 uint32_t numRois = getSizeOfDimension(roiShape, 0); in prepare() local
218 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
220 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
231 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
339 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass() argument
347 for (uint32_t b = 0; b < numRois; b++) { in hardNmsMultiClass()
403 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass() argument
411 for (uint32_t b = 0; b < numRois; b++) { in softNmsMultiClass()
457 uint32_t numRois = getSizeOfDimension(scoresShape, 0); in boxWithNmsLimitFloat32Compute() local
463 for (uint32_t i = 0; i < numRois; i++) { in boxWithNmsLimitFloat32Compute()
681 uint32_t numRois = getSizeOfDimension(scoreShape, 0); in prepare() local
683 NN_RET_CHECK(getSizeOfDimension(roiShape, 0) == numRois); in prepare()
685 NN_RET_CHECK(getSizeOfDimension(batchesShape, 0) == numRois); in prepare()