Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc168 TfLiteTensor* detection_classes = in Prepare() local
170 detection_classes->type = kTfLiteFloat32; in Prepare()
171 SetTensorSizes(context, detection_classes, {kBatchSize, num_detected_boxes}); in Prepare()
462 TfLiteTensor* detection_classes = in NonMaxSuppressionMultiClassRegularHelper() local
551 detection_classes->data.f[output_box_index] = class_index; in NonMaxSuppressionMultiClassRegularHelper()
558 detection_classes->data.f[output_box_index] = 0.0f; in NonMaxSuppressionMultiClassRegularHelper()
587 TfLiteTensor* detection_classes = in NonMaxSuppressionMultiClassFastHelper() local
635 detection_classes->data.f[box_offset] = class_indices[col]; in NonMaxSuppressionMultiClassFastHelper()