Searched refs:num_detections_per_class (Results 1 – 1 of 1) sorted by relevance
511 int num_detections_per_class; member528 selected.reserve(nms_task_param.num_detections_per_class); in ComputeNMSResult()545 class_scores, nms_task_param.num_detections_per_class, &selected)); in ComputeNMSResult()578 sorted_box_info.resize(nms_task_param.num_detections_per_class + in Run()638 const int num_detections_per_class = in NonMaxSuppressionMultiClassRegularHelper() local645 TF_LITE_ENSURE(context, num_detections_per_class > 0); in NonMaxSuppressionMultiClassRegularHelper()649 max_detections + num_detections_per_class); in NonMaxSuppressionMultiClassRegularHelper()660 num_detections_per_class, in NonMaxSuppressionMultiClassRegularHelper()