Revert "update the tensorflow.lite to use ai-edge-litert for all pyth…

…on based…" (#2705)

Reverts #2702

BUG=#2703

python/tflite_micro/BUILD

@@ -94,7 +94,6 @@ py_test(
     ],
     deps = [
         ":runtime",
-        requirement("ai-edge-litert"),
         requirement("numpy"),
         requirement("tensorflow"),
         "//tensorflow/lite/micro/examples/recipes:add_four_numbers",

python/tflite_micro/runtime_test.py

@@ -25,7 +25,6 @@
 import numpy as np
 import tensorflow as tf
 
-from ai_edge_litert import interpreter as litert_interpreter
 from tensorflow.python.framework import test_util
 from tensorflow.python.platform import test
 from tflite_micro.python.tflite_micro import runtime
@@ -200,10 +199,10 @@ def testCompareWithTFLite(self):
     tflm_interpreter = runtime.Interpreter.from_bytes(model_data)
 
     # TFLite interpreter
-    tflite_interpreter = litert_interpreter.Interpreter(
+    tflite_interpreter = tf.lite.Interpreter(
         model_content=model_data,
         experimental_op_resolver_type=\
-        litert_interpreter.OpResolverType.BUILTIN_REF)
+        tf.lite.experimental.OpResolverType.BUILTIN_REF)
     tflite_interpreter.allocate_tensors()
     tflite_output_details = tflite_interpreter.get_output_details()[0]
     tflite_input_details = tflite_interpreter.get_input_details()[0]

tensorflow/lite/micro/examples/hello_world/BUILD

@@ -53,7 +53,6 @@ py_binary(
         "@absl_py//absl:app",
         "@absl_py//absl/flags",
         "@absl_py//absl/logging",
-        requirement("ai-edge-litert"),
         requirement("numpy"),
         requirement("tensorflow"),
         "//python/tflite_micro:runtime",

tensorflow/lite/micro/examples/hello_world/evaluate.py

@@ -16,7 +16,6 @@
 import tensorflow as tf
 from absl import app
 from absl import flags
-from ai_edge_litert import interpreter as litert_interpreter
 import numpy as np
 import matplotlib.pyplot as plt
 from tensorflow.python.platform import resource_loader
@@ -93,9 +92,9 @@ def get_tflm_prediction(model_path, x_values):
 # returns the prediction of the interpreter.
 def get_tflite_prediction(model_path, x_values):
   # TFLite interpreter
-  tflite_interpreter = litert_interpreter.Interpreter(
+  tflite_interpreter = tf.lite.Interpreter(
       model_path=model_path,
-      experimental_op_resolver_type=litert_interpreter.OpResolverType.
+      experimental_op_resolver_type=tf.lite.experimental.OpResolverType.
       BUILTIN_REF,
   )
   tflite_interpreter.allocate_tensors()

tensorflow/lite/micro/examples/mnist_lstm/BUILD

@@ -6,7 +6,6 @@ py_binary(
     srcs = ["train.py"],
     srcs_version = "PY3",
     deps = [
-        requirement("ai-edge-litert"),
         requirement("numpy"),
         requirement("tensorflow"),
     ],

tensorflow/lite/micro/examples/mnist_lstm/evaluate_test.py

@@ -17,7 +17,6 @@
 import numpy as np
 import tensorflow as tf
 
-from ai_edge_litert import interpreter as litert_interpreter
 from tensorflow.python.framework import test_util
 from tensorflow.python.platform import resource_loader
 from tensorflow.python.platform import test
@@ -44,10 +43,10 @@ def testInputErrHandling(self):
       evaluate.predict_image(self.tflm_interpreter, wrong_size_image_path)
 
   def testCompareWithTFLite(self):
-    tflite_interpreter = litert_interpreter.Interpreter(
+    tflite_interpreter = tf.lite.Interpreter(
         model_path=self.model_path,
         experimental_op_resolver_type=\
-        litert_interpreter.OpResolverType.BUILTIN_REF)
+        tf.lite.experimental.OpResolverType.BUILTIN_REF)
     tflite_interpreter.allocate_tensors()
     tflite_output_details = tflite_interpreter.get_output_details()[0]
     tflite_input_details = tflite_interpreter.get_input_details()[0]

tensorflow/lite/micro/tools/BUILD

@@ -34,7 +34,6 @@ py_library(
     srcs_version = "PY3",
     visibility = ["//:__subpackages__"],
     deps = [
-        requirement("ai-edge-litert"),
         "//tensorflow/lite/python:schema_py",
     ],
 )
@@ -209,7 +208,6 @@ py_binary(
         ":model_transforms_utils",
         "@absl_py//absl:app",
         "@absl_py//absl/flags",
-        requirement("ai-edge-litert"),
         requirement("tensorflow"),
         "//python/tflite_micro:runtime",
         "//tensorflow/lite/tools:flatbuffer_utils",

tensorflow/lite/micro/tools/generate_test_for_model.py

@@ -18,7 +18,6 @@
 import numpy as np
 import tensorflow as tf
 
-from ai_edge_litert import interpreter as litert_interpreter
 from tflite_micro.tensorflow.lite.python import schema_py_generated as schema_fb
 
 
@@ -104,9 +103,9 @@ def generate_golden_single_in_single_out(self):
     if (len(self.model_paths) != 1):
       raise RuntimeError(f'Single model expected')
     model_path = self.model_paths[0]
-    interpreter = litert_interpreter.Interpreter(model_path=model_path,
+    interpreter = tf.lite.Interpreter(model_path=model_path,
                                       experimental_op_resolver_type=\
-                                      litert_interpreter.OpResolverType.BUILTIN_REF)
+                                      tf.lite.experimental.OpResolverType.BUILTIN_REF)
 
     interpreter.allocate_tensors()
 
@@ -141,10 +140,10 @@ def generate_goldens(self, builtin_operator):
 
     for model_path in self.model_paths:
       # Load model and run a single inference with random inputs.
-      interpreter = litert_interpreter.Interpreter(
+      interpreter = tf.lite.Interpreter(
           model_path=model_path,
           experimental_op_resolver_type=\
-          litert_interpreter.OpResolverType.BUILTIN_REF)
+          tf.lite.experimental.OpResolverType.BUILTIN_REF)
       interpreter.allocate_tensors()
       input_tensor = interpreter.tensor(
           interpreter.get_input_details()[0]['index'])

tensorflow/lite/micro/tools/layer_by_layer_debugger.py

@@ -20,7 +20,6 @@
 from absl import app
 from absl import flags
 from absl import logging
-from ai_edge_litert import interpreter as litert_interpreter
 import numpy as np
 import tensorflow as tf
 
@@ -195,7 +194,7 @@ def main(_) -> None:
       intrepreter_config=runtime.InterpreterConfig.kPreserveAllTensors,
   )
 
-  tflite_interpreter = litert_interpreter.Interpreter(
+  tflite_interpreter = tf.lite.Interpreter(
       model_path=_INPUT_TFLITE_FILE.value,
       experimental_preserve_all_tensors=True,
   )

third_party/python_requirements.in

@@ -34,4 +34,3 @@ mako
 pillow
 yapf
 protobuf
-ai-edge-litert

third_party/python_requirements.txt

@@ -11,9 +11,6 @@ absl-py==2.0.0 \
     #   keras
     #   tensorboard
     #   tensorflow
-ai-edge-litert==1.0.1 \
-    --hash=sha256:25a9b1577941498842bf77630722eda1163026c37abd57af66791a6955551b9d
-    # via -r third_party/python_requirements.in
 astunparse==1.6.3 \
     --hash=sha256:5ad93a8456f0d084c3456d059fd9a92cce667963232cbf763eac3bc5b7940872 \
     --hash=sha256:c2652417f2c8b5bb325c885ae329bdf3f86424075c4fd1a128674bc6fba4b8e8
@@ -508,7 +505,6 @@ numpy==1.26.3 \
     --hash=sha256:f73497e8c38295aaa4741bdfa4fda1a5aedda5473074369eca10626835445511
     # via
     #   -r third_party/python_requirements.in
-    #   ai-edge-litert
     #   h5py
     #   keras
     #   ml-dtypes