Searched refs:class_idx (Results 1 – 2 of 2) sorted by relevance
259 int class_idx; in BatchedNonMaxSuppressionOp() member269 for (int class_idx = 0; class_idx < num_classes; ++class_idx) { in BatchedNonMaxSuppressionOp() local278 class_scores_data.push_back(scores_data[box * num_classes + class_idx]); in BatchedNonMaxSuppressionOp()283 boxes_data[(box * q + class_idx) * 4 + cid]); in BatchedNonMaxSuppressionOp()344 class_idx, in BatchedNonMaxSuppressionOp()391 nmsed_classes[batch].push_back(next_candidate.class_idx); in BatchedNonMaxSuppressionOp()
3 DEBUG_LOCKS_WARN_ON(class_idx > MAX_LOCKDEP_KEYS)