commit 9a43ab39f2db65d5526773c9c6b45f2087e4c1c7
author Sachin Joglekar <srjoglekar@google.com> Tue May 12 15:41:09 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Tue May 12 15:44:28 2020 -0700
tree ceabb3bcee970e05f14e7af7749ac52ca8b214ad
parent f38355dab31bb466e9fdc900089dcd4abba536d6

Modify op version in optimize only if convertor version < quantized version.

PiperOrigin-RevId: 311216743
Change-Id: Iaac04750d0d302e9bba11b223c2885d6a36d74b3

tensorflow/lite/tools/optimize/model_utils.cc

diff --git a/tensorflow/lite/tools/optimize/model_utils.cc b/tensorflow/lite/tools/optimize/model_utils.cc
index 26dcff2..ae868cf 100644
--- a/tensorflow/lite/tools/optimize/model_utils.cc
+++ b/tensorflow/lite/tools/optimize/model_utils.cc
@@ -134,8 +134,10 @@
       OperatorCodeT* op_code = model->operator_codes[op->opcode_index].get();
       operator_property::OperatorProperty property =
           operator_property::GetOperatorProperty(model, subgraph_idx, op_idx);
-      if (property.quantizable) {
-        // Only update the versions of quantizable operations.
+      if (property.quantizable && op_code->version < property.version) {
+        // Only update the versions of quantizable operations if the original
+        // version is lesser than minimum quantized one mentioned by
+        // OperatorProperty.
         op_code->version = property.version;
       }
     }