Searched refs:nms_task_param (Results 1 – 1 of 1) sorted by relevance
523 TfLiteStatus ComputeNMSResult(const NMSTaskParam& nms_task_param, int col_begin, in ComputeNMSResult() argument526 std::vector<float> class_scores(nms_task_param.num_boxes); in ComputeNMSResult()528 selected.reserve(nms_task_param.num_detections_per_class); in ComputeNMSResult()532 nms_task_param.scores + col + nms_task_param.label_offset; in ComputeNMSResult()533 for (int row = 0; row < nms_task_param.num_boxes; row++) { in ComputeNMSResult()536 scores_base += nms_task_param.num_classes_with_background; in ComputeNMSResult()542 nms_task_param.context, in ComputeNMSResult()544 nms_task_param.context, nms_task_param.node, nms_task_param.op_data, in ComputeNMSResult()545 class_scores, nms_task_param.num_detections_per_class, &selected)); in ComputeNMSResult()554 (selected[i] * nms_task_param.num_classes_with_background + col + in ComputeNMSResult()[all …]