vee1e/tflite-micro
1
0
Fork 0
mirror of https://github.com/vee1e/tflite-micro.git synced 2026-09-01 17:57:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert "update the tensorflow.lite to use ai-edge-litert for all python based…" (#2705)

Browse source 
Reverts tensorflow/tflite-micro#2702

BUG=#2703
This commit is contained in:
RJ Ascani 2024-09-27 10:44:05 -07:00 • committed by GitHub
parent 47b5450743
commit 6411584e33
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 11 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

5
python/tflite_micro/runtime_test.py
View file

@ -25,7 +25,6 @@ import weakref
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 @@ class ConvModelTests(test_util.TensorFlowTestCase):
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]