Searched refs:input_class_predictions (Results 1 – 1 of 1) sorted by relevance
146 const TfLiteTensor* input_class_predictions = in Prepare() local151 TF_LITE_ENSURE_EQ(context, NumDimensions(input_class_predictions), 3); in Prepare()205 {input_class_predictions->dims->data[1], in Prepare()206 input_class_predictions->dims->data[2]}); in Prepare()645 void DequantizeClassPredictions(const TfLiteTensor* input_class_predictions, in DequantizeClassPredictions() argument650 static_cast<float>(input_class_predictions->params.zero_point); in DequantizeClassPredictions()651 float quant_scale = static_cast<float>(input_class_predictions->params.scale); in DequantizeClassPredictions()653 const uint8* scores_quant = GetTensorData<uint8>(input_class_predictions); in DequantizeClassPredictions()664 const TfLiteTensor* input_class_predictions = in NonMaxSuppressionMultiClass() local668 TF_LITE_ENSURE_EQ(context, input_class_predictions->dims->data[0], in NonMaxSuppressionMultiClass()[all …]