Searched refs:keep_scores (Results 1 – 1 of 1) sorted by relevance
398 std::vector<float> keep_scores; in NonMaxSuppressionSingleClassHelper() local400 scores, non_max_suppression_score_threshold, &keep_scores, &keep_indices); in NonMaxSuppressionSingleClassHelper()402 int num_scores_kept = keep_scores.size(); in NonMaxSuppressionSingleClassHelper()405 DecreasingPartialArgSort(keep_scores.data(), num_scores_kept, num_scores_kept, in NonMaxSuppressionSingleClassHelper()