Lines Matching refs:select
312 std::function<const float*(uint32_t)> getRoiBase, uint32_t* select, in hardNmsSingleClass() argument
314 uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0; in hardNmsSingleClass()
343 std::vector<uint32_t>* select) { in hardNmsMultiClass() argument
346 uint32_t size = select->size(); in hardNmsMultiClass()
351 select->push_back(index); in hardNmsMultiClass()
354 uint32_t* selectStart = select->data() + size; in hardNmsMultiClass()
355 uint32_t selectLength = select->size() - size; in hardNmsMultiClass()
358 select->resize(selectEnd - select->data()); in hardNmsMultiClass()
362 std::sort(select->begin(), select->end(), in hardNmsMultiClass()
366 if (maxNumDetections < 0 || select->size() <= static_cast<size_t>(maxNumDetections)) { in hardNmsMultiClass()
369 select->resize(maxNumDetections); in hardNmsMultiClass()
376 uint32_t* select, uint32_t selectLength) { in softNmsSingleClass() argument
377 uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0; in softNmsSingleClass()
407 std::vector<uint32_t>* select) { in softNmsMultiClass() argument
410 uint32_t size = select->size(); in softNmsMultiClass()
415 select->push_back(index); in softNmsMultiClass()
418 uint32_t* selectStart = select->data() + size; in softNmsMultiClass()
419 uint32_t selectLength = select->size() - size; in softNmsMultiClass()
423 select->resize(selectEnd - select->data()); in softNmsMultiClass()
427 std::sort(select->begin(), select->end(), in softNmsMultiClass()
431 if (maxNumDetections < 0 || select->size() <= static_cast<size_t>(maxNumDetections)) { in softNmsMultiClass()
434 select->resize(maxNumDetections); in softNmsMultiClass()
828 std::vector<uint32_t>* select) { in filterBoxes() argument
831 for (uint32_t j = 0; j < select->size(); j++) { in filterBoxes()
832 const float* roiInfo = roiBase + (*select)[j] * kRoiDim; in filterBoxes()
840 (*select)[i++] = (*select)[j]; in filterBoxes()
843 select->resize(i); in filterBoxes()
911 std::vector<uint32_t> select(batchSize); in generateProposalsNhwcFloat32Compute() local
912 std::iota(select.begin(), select.end(), 0); in generateProposalsNhwcFloat32Compute()
913 if (preNmsTopN > 0 && static_cast<size_t>(preNmsTopN) < select.size()) { in generateProposalsNhwcFloat32Compute()
914 std::sort(select.begin(), select.end(), in generateProposalsNhwcFloat32Compute()
918 select.resize(preNmsTopN); in generateProposalsNhwcFloat32Compute()
922 filterBoxes(roiTransformedBuffer.data(), imageInfoBase, minSize, &select); in generateProposalsNhwcFloat32Compute()
930 select.data(), select.size()); in generateProposalsNhwcFloat32Compute()
931 uint32_t selectSize = selectEnd - select.data(); in generateProposalsNhwcFloat32Compute()
932 select.resize(selectSize); in generateProposalsNhwcFloat32Compute()
935 for (auto i : select) { in generateProposalsNhwcFloat32Compute()
1297 std::vector<uint32_t> select; in detectionPostprocessFloat32() local
1304 &select); in detectionPostprocessFloat32()
1305 for (uint32_t i = 0; i < select.size(); i++) { in detectionPostprocessFloat32()
1306 uint32_t ind = select[i]; in detectionPostprocessFloat32()
1312 *detectionOutData++ = select.size(); in detectionPostprocessFloat32()
1320 std::vector<uint32_t> select; in detectionPostprocessFloat32() local
1323 select.push_back(a); in detectionPostprocessFloat32()
1329 select.data(), select.size()); in detectionPostprocessFloat32()
1330 select.resize(selectEnd - select.data()); in detectionPostprocessFloat32()
1334 for (auto i : select) { in detectionPostprocessFloat32()
1349 *detectionOutData++ = select.size() * numOutClasses; in detectionPostprocessFloat32()