Internal change

PiperOrigin-RevId: 356096456
Change-Id: Ifd31de7a38731e942a1f9fa0a0c31b5cf7842f1f

tensorflow/lite/python/convert.py

@@ -322,6 +322,7 @@ def build_toco_flags(inference_type=dtypes.float32,
                      target_ops=None,
                      conversion_summary_dir=None,
                      select_user_tf_ops=None,
+                     enable_tflite_resource_variables=False,
                      **_):
   """Build the TOCO flags object from params."""
   toco = _toco_flags_pb2.TocoFlags()
@@ -355,6 +356,7 @@ def build_toco_flags(inference_type=dtypes.float32,
       toco.enable_select_tf_ops = True
     if set(target_ops) == set([OpsSet.SELECT_TF_OPS]):
       toco.force_select_tf_ops = True
+  toco.enable_tflite_resource_variables = enable_tflite_resource_variables
   return toco
 
 

tensorflow/lite/python/lite.py

@@ -720,6 +720,7 @@ class TFLiteSavedModelConverterV2(TFLiteConverterBaseV2):
     self._saved_model_exported_names = saved_model_exported_names
     self._trackable_obj = trackable_obj
     self._parse_saved_model_args(always_enable_saved_model_import=True)
+    self._enable_tflite_resource_variables = False
 
   def convert(self):
     """Converts a TensorFlow GraphDef based on instance variables.
@@ -779,6 +780,10 @@ class TFLiteSavedModelConverterV2(TFLiteConverterBaseV2):
 
     converter_kwargs = self._get_base_converter_args()
     converter_kwargs.update(quant_mode.converter_flags())
+    converter_kwargs.update({
+        "enable_tflite_resource_variables":
+            self._enable_tflite_resource_variables
+    })
 
     result = _convert_saved_model(**converter_kwargs)
     calibrate_and_quantize, flags = quant_mode.quantizer_flags()