Home
last modified time | relevance | path

Searched refs:keep_indices (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc329 std::vector<int>* keep_indices) { in SelectDetectionsAboveScoreThreshold() argument
333 keep_indices->emplace_back(i); in SelectDetectionsAboveScoreThreshold()
395 std::vector<int> keep_indices; in NonMaxSuppressionSingleClassHelper() local
400 scores, non_max_suppression_score_threshold, &keep_scores, &keep_indices); in NonMaxSuppressionSingleClassHelper()
422 selected->push_back(keep_indices[sorted_indices[i]]); in NonMaxSuppressionSingleClassHelper()
431 decoded_boxes, keep_indices[sorted_indices[i]], in NonMaxSuppressionSingleClassHelper()
432 keep_indices[sorted_indices[j]]); in NonMaxSuppressionSingleClassHelper()