diff --git a/tensorflow/lite/tools/optimize/quantize_model.cc b/tensorflow/lite/tools/optimize/quantize_model.cc index 30d9f961098..42db16eb965 100644 --- a/tensorflow/lite/tools/optimize/quantize_model.cc +++ b/tensorflow/lite/tools/optimize/quantize_model.cc @@ -300,7 +300,7 @@ TfLiteStatus ApplyConstraints(ModelT* model, if (!property.quantizable) { continue; } - if ((!property.arbitrary_inputs) || + if (!property.arbitrary_inputs || !property.restrict_same_input_output_scale) { continue; }