Searched refs:detection_classes (Results 1 – 1 of 1) sorted by relevance
168 TfLiteTensor* detection_classes = in Prepare() local170 detection_classes->type = kTfLiteFloat32; in Prepare()171 SetTensorSizes(context, detection_classes, {kBatchSize, num_detected_boxes}); in Prepare()462 TfLiteTensor* detection_classes = in NonMaxSuppressionMultiClassRegularHelper() local551 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() local635 detection_classes->data.f[box_offset] = class_indices[col]; in NonMaxSuppressionMultiClassFastHelper()