diff --git a/mediapipe/calculators/tflite/tflite_inference_calculator_test_common.h b/mediapipe/calculators/tflite/tflite_inference_calculator_test_common.h index cf995f47b..19d46d535 100644 --- a/mediapipe/calculators/tflite/tflite_inference_calculator_test_common.h +++ b/mediapipe/calculators/tflite/tflite_inference_calculator_test_common.h @@ -72,6 +72,9 @@ void DoSmokeTest(const std::string& graph_proto) { affine_quant->zero_point->data[0] = 0; quant.type = kTfLiteAffineQuantization; quant.params = affine_quant; + } else { + quant.type = kTfLiteNoQuantization; + quant.params = nullptr; } interpreter->SetTensorParametersReadWrite(0, tflite::typeToTfLiteType(), "", {3}, quant);