Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc146 const TfLiteTensor* input_class_predictions = in Prepare() local
151 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() argument
650 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() local
668 TF_LITE_ENSURE_EQ(context, input_class_predictions->dims->data[0], in NonMaxSuppressionMultiClass()
[all …]