Lines Matching refs:numClasses

67     uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / roiLength;  in bboxTransformFloat32()  local
89 for (uint32_t i = 0; i < numClasses; i++) { in bboxTransformFloat32()
244 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / kRoiDim; in prepare() local
246 NN_RET_CHECK_GT(numClasses, 0); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 1), kRoiDim * numClasses); in prepare()
262 outputShape.dimensions = {numRois, numClasses * kRoiDim}; in prepare()
387 void hardNmsMultiClass(const float* scoresData, uint32_t numClasses, uint32_t numRois, in hardNmsMultiClass() argument
393 for (uint32_t c = 1; c < numClasses; c++) { in hardNmsMultiClass()
396 const uint32_t index = b * numClasses + c; in hardNmsMultiClass()
451 void softNmsMultiClass(float* scoresData, uint32_t numClasses, uint32_t numRois, in softNmsMultiClass() argument
457 for (uint32_t c = 1; c < numClasses; c++) { in softNmsMultiClass()
460 const uint32_t index = b * numClasses + c; in softNmsMultiClass()
506 uint32_t numClasses = getSizeOfDimension(scoresShape, 1); in boxWithNmsLimitFloat32Compute() local
531 scoresBase, numClasses, batchSplitIn->at(b), scoreThreshold, nmsScoreThreshold, in boxWithNmsLimitFloat32Compute()
536 [&scoresBase, numClasses](const uint32_t& lhs, const uint32_t& rhs) { in boxWithNmsLimitFloat32Compute()
537 uint32_t lhsClass = lhs % numClasses, rhsClass = rhs % numClasses; in boxWithNmsLimitFloat32Compute()
543 scoresBase += batchSplitIn->at(b) * numClasses; in boxWithNmsLimitFloat32Compute()
544 roiBase += batchSplitIn->at(b) * numClasses * kRoiDim; in boxWithNmsLimitFloat32Compute()
571 uint32_t numClasses = getSizeOfDimension(scoresShape, 1); in boxWithNmsLimitWriteOutput() local
607 *classesOutPtr++ = index % numClasses; in boxWithNmsLimitWriteOutput()
610 scoresBase += batchSplitIn[b] * numClasses; in boxWithNmsLimitWriteOutput()
611 roiBase += batchSplitIn[b] * numClasses * kRoiDim; in boxWithNmsLimitWriteOutput()
770 uint32_t numClasses = getSizeOfDimension(scoreShape, 1); in prepare() local
772 NN_RET_CHECK(getSizeOfDimension(roiShape, 1) == kRoiDim * numClasses); in prepare()
774 NN_RET_CHECK_GT(numClasses, 1); in prepare()
1451 uint32_t numClasses = getSizeOfDimension(scoreShape, 2); in detectionPostprocessFloat32() local
1485 scoreBase, numClasses, numAnchors, scoreThreshold, iouThreshold, in detectionPostprocessFloat32()
1487 [&roiBuffer, numClasses](uint32_t ind) { in detectionPostprocessFloat32()
1488 return roiBuffer.data() + (ind / numClasses) * kRoiDim; in detectionPostprocessFloat32()
1494 memcpy(roiOutBase + i * kRoiDim, &roiBuffer[(ind / numClasses) * kRoiDim], in detectionPostprocessFloat32()
1496 classOutBase[i] = (ind % numClasses) - (isBGInLabel ? 0 : 1); in detectionPostprocessFloat32()
1500 uint32_t numOutClasses = std::min<uint32_t>(numClasses - 1, maxClassesPerDetection); in detectionPostprocessFloat32()
1503 maxScores[a] = *std::max_element(scoreBase + a * numClasses + 1, in detectionPostprocessFloat32()
1504 scoreBase + (a + 1) * numClasses); in detectionPostprocessFloat32()
1521 const float* score = scoreBase + i * numClasses; in detectionPostprocessFloat32()
1522 std::vector<uint32_t> scoreInds(numClasses - 1); in detectionPostprocessFloat32()
1537 scoreBase += numAnchors * numClasses; in detectionPostprocessFloat32()
1622 uint32_t numClasses = getSizeOfDimension(scoreShape, 2); in prepare() local
1651 NN_RET_CHECK_GT(numClasses, 1); in prepare()