Searched refs:max_detections (Results 1 – 3 of 3) sorted by relevance
81 int max_detections; member99 op_data->max_detections = m["max_detections"].AsInt32(); in Init()155 op_data->max_detections * op_data->max_classes_per_detection; in Prepare()375 int max_detections) { in NonMaxSuppressionSingleClassHelper() argument386 TF_LITE_ENSURE(context, (max_detections >= 0)); in NonMaxSuppressionSingleClassHelper()408 const int output_size = std::min(num_boxes_kept, max_detections); in NonMaxSuppressionSingleClassHelper()472 const int max_detections = op_data->max_detections; in NonMaxSuppressionMultiClassRegularHelper() local483 num_boxes + max_detections); in NonMaxSuppressionMultiClassRegularHelper()485 max_detections); in NonMaxSuppressionMultiClassRegularHelper()489 sorted_indices.resize(max_detections); in NonMaxSuppressionMultiClassRegularHelper()[all …]
58 A [batch_size, max_detections, 4] float32 tensor 65 A [batch_size, max_detections] float32 tensor 72 A [batch_size, max_detections] float32 tensor
357 int max_detections = 0; in BatchedNonMaxSuppressionOp() local360 max_detections = in BatchedNonMaxSuppressionOp()366 max_detections = in BatchedNonMaxSuppressionOp()370 final_valid_detections.push_back(max_detections); in BatchedNonMaxSuppressionOp()372 int curr_total_size = max_detections; in BatchedNonMaxSuppressionOp()