Internal change
PiperOrigin-RevId: 356096456
Change-Id: Ifd31de7a38731e942a1f9fa0a0c31b5cf7842f1f
diff --git a/tensorflow/lite/python/convert.py b/tensorflow/lite/python/convert.py
index 42fbf23..71ef18e 100644
--- a/tensorflow/lite/python/convert.py
+++ b/tensorflow/lite/python/convert.py
@@ -322,6 +322,7 @@
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 @@
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
diff --git a/tensorflow/lite/python/lite.py b/tensorflow/lite/python/lite.py
index fc2f944..f159ee9 100644
--- a/tensorflow/lite/python/lite.py
+++ b/tensorflow/lite/python/lite.py
@@ -720,6 +720,7 @@
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 @@
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()
