non_max_suppression_op.cc | 259 int class_idx; member in struct:tensorflow::__anon45437::ResultCandidate 269 for (int class_idx = 0; class_idx < num_classes; ++class_idx) { 278 class_scores_data.push_back(scores_data[box * num_classes + class_idx]); 283 boxes_data[(box * q + class_idx) * 4 + cid]); 344 class_idx, 391 nmsed_classes[batch].push_back(next_candidate.class_idx);
|