cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 519805707
This commit is contained in:
MediaPipe Team 2023-03-27 13:00:37 -07:00 committed by Copybara-Service
parent e2f2acca5e
commit 1fdc82d7ec
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mediapipe/calculators/tflite/tflite_inference_calculator_test_common.h
View File

@ -72,6 +72,9 @@ void DoSmokeTest(const std::string& graph_proto) {
affine_quant->zero_point->data[0] = 0; affine_quant->zero_point->data[0] = 0;
quant.type = kTfLiteAffineQuantization; quant.type = kTfLiteAffineQuantization;
quant.params = affine_quant; quant.params = affine_quant;
} else {
quant.type = kTfLiteNoQuantization;
quant.params = nullptr;
} }
interpreter->SetTensorParametersReadWrite(0, tflite::typeToTfLiteType<T>(), interpreter->SetTensorParametersReadWrite(0, tflite::typeToTfLiteType<T>(),
"", {3}, quant); "", {3}, quant);