Home
last modified time | relevance | path

Searched refs:detection_scores (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc174 TfLiteTensor* detection_scores = in Prepare() local
176 detection_scores->type = kTfLiteFloat32; in Prepare()
177 SetTensorSizes(context, detection_scores, {kBatchSize, num_detected_boxes}); in Prepare()
464 TfLiteTensor* detection_scores = in NonMaxSuppressionMultiClassRegularHelper() local
553 detection_scores->data.f[output_box_index] = selected_score; in NonMaxSuppressionMultiClassRegularHelper()
560 detection_scores->data.f[output_box_index] = 0.0f; in NonMaxSuppressionMultiClassRegularHelper()
589 TfLiteTensor* detection_scores = in NonMaxSuppressionMultiClassFastHelper() local
637 detection_scores->data.f[box_offset] = box_scores[class_indices[col]]; in NonMaxSuppressionMultiClassFastHelper()