Lines Matching refs:selectStart

365     uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0;  in hardNmsSingleClass()  local
369 while (selectStart < selectEnd && numDetections < maxNumDetections) { in hardNmsSingleClass()
371 auto& maxScore = *std::max_element(selectStart, selectEnd, in hardNmsSingleClass()
375 std::swap(maxScore, *selectStart); in hardNmsSingleClass()
378 for (uint32_t* i = selectStart + 1; i < selectEnd; i++) { in hardNmsSingleClass()
379 float iou = getIoUAxisAligned(getRoiBase(*i), getRoiBase(*selectStart)); in hardNmsSingleClass()
384 selectStart++; in hardNmsSingleClass()
387 return selectStart; in hardNmsSingleClass()
405 uint32_t* selectStart = select->data() + size; in hardNmsMultiClass() local
408 getRoiBase, selectStart, selectLength); in hardNmsMultiClass()
428 uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0; in softNmsSingleClass() local
432 while (selectStart < selectEnd && numDetections < maxNumDetections) { in softNmsSingleClass()
434 auto& maxScore = *std::max_element(selectStart, selectEnd, in softNmsSingleClass()
438 std::swap(maxScore, *selectStart); in softNmsSingleClass()
441 for (uint32_t* i = selectStart + 1; i < selectEnd; i++) { in softNmsSingleClass()
442 float iou = getIoUAxisAligned(getRoiBase(*i), getRoiBase(*selectStart)); in softNmsSingleClass()
448 selectStart++; in softNmsSingleClass()
451 return selectStart; in softNmsSingleClass()
469 uint32_t* selectStart = select->data() + size; in softNmsMultiClass() local
473 getRoiBase, kernel, selectStart, selectLength); in softNmsMultiClass()