Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dnon_max_suppression_op.cc184 Candidate next_candidate; in DoNonMaxSuppressionOp() local
187 next_candidate = candidate_priority_queue.top(); in DoNonMaxSuppressionOp()
196 if (suppress_check_fn(next_candidate.box_index, selected[j])) { in DoNonMaxSuppressionOp()
203 selected.push_back(next_candidate.box_index); in DoNonMaxSuppressionOp()
204 selected_scores.push_back(next_candidate.score); in DoNonMaxSuppressionOp()
315 Candidate next_candidate; in BatchedNonMaxSuppressionOp() local
324 next_candidate = candidate_vector[candidate_idx++]; in BatchedNonMaxSuppressionOp()
331 if (IOUGreaterThanThreshold(boxes_data, next_candidate.box_index, in BatchedNonMaxSuppressionOp()
339 selected.push_back(next_candidate.box_index); in BatchedNonMaxSuppressionOp()
341 int id = next_candidate.box_index; in BatchedNonMaxSuppressionOp()
[all …]