Searched refs:sorted_indices (Results 1 – 4 of 4) sorted by relevance
403 std::vector<int> sorted_indices; in NonMaxSuppressionSingleClassHelper() local404 sorted_indices.resize(num_scores_kept); in NonMaxSuppressionSingleClassHelper()406 sorted_indices.data()); 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()488 std::vector<int> sorted_indices; in NonMaxSuppressionMultiClassRegularHelper() local489 sorted_indices.resize(max_detections); in NonMaxSuppressionMultiClassRegularHelper()518 sorted_indices.data()); in NonMaxSuppressionMultiClassRegularHelper()522 int temp = sorted_indices[row]; in NonMaxSuppressionMultiClassRegularHelper()[all …]
88 Tensor sorted_indices(tensorflow::DT_INT32, {how_many_labels}); in GetTopLabels() local91 sorted_indices.flat<int>()(i) = scores[i].first; in GetTopLabels()94 *out_indices = sorted_indices; in GetTopLabels()
244 Tensor sorted_indices(tensorflow::DT_INT32, {scores.size()}); in GetTopLabels() local247 sorted_indices.flat<int>()(i) = scores[i].first; in GetTopLabels()250 *out_indices = sorted_indices; in GetTopLabels()
272 Tensor sorted_indices(tensorflow::DT_INT32, {scores.size()}); in GetTopLabels() local275 sorted_indices.flat<int>()(i) = scores[i].first; in GetTopLabels()278 *out_indices = sorted_indices; in GetTopLabels()