Internal change
PiperOrigin-RevId: 519805707
This commit is contained in:
parent
e2f2acca5e
commit
1fdc82d7ec
|
@ -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<T>(),
|
||||
"", {3}, quant);
|
||||
|
|
Loading…
Reference in New Issue
Block a user