Searched refs:class_scores_data (Results 1 – 1 of 1) sorted by relevance
270 std::vector<float> class_scores_data; in BatchedNonMaxSuppressionOp() local271 class_scores_data.reserve(num_boxes); in BatchedNonMaxSuppressionOp()278 class_scores_data.push_back(scores_data[box * num_classes + class_idx]); in BatchedNonMaxSuppressionOp()307 for (int i = 0; i < class_scores_data.size(); ++i) { in BatchedNonMaxSuppressionOp()308 if (class_scores_data[i] > score_threshold) { in BatchedNonMaxSuppressionOp()309 candidate_vector.emplace_back(Candidate({i, class_scores_data[i]})); in BatchedNonMaxSuppressionOp()