Searched refs:class_idx (Results 1 – 3 of 3) sorted by relevance
81 for (int class_idx = 0; class_idx < num_classes; ++class_idx) {82 T pred = predictions(batch_idx, class_idx);
300 int class_idx; member304 void DoNMSPerClass(int batch_idx, int class_idx, const float* boxes_data, in DoNMSPerClass() argument315 class_scores_data.push_back(scores_data[box_idx * num_classes + class_idx]); in DoNMSPerClass()319 boxes_data[(box_idx * q + class_idx) * 4 + cid]); in DoNMSPerClass()371 result_candidate_vec[selected.size() + size_per_class * class_idx] = { in DoNMSPerClass()374 class_idx, in DoNMSPerClass()432 nmsed_classes.push_back(next_candidate.class_idx); in SelectResultPerBatch()474 int class_idx = idx % num_classes; in BatchedNonMaxSuppressionOp() local475 DoNMSPerClass(batch_idx, class_idx, in BatchedNonMaxSuppressionOp()
5944 const int class_idx = 1 + offset; in ConvertCombinedNMS() local5945 const int num_classes = scores_dims.d[class_idx]; in ConvertCombinedNMS()5948 boxes_dims.d[class_idx] == 1 || boxes_dims.d[class_idx] == num_classes; in ConvertCombinedNMS()6007 bool share_location = (boxes_dims.d[class_idx] == 1); in ConvertCombinedNMS()