intermediate results in tflite

Signed-off-by: xavier dupré <[email protected]>

Author: sdpython

tests/tfhub/_tools.py

@@ -403,9 +403,15 @@ def call_tflite(inp):
         with open(onnx_name, "rb") as f:
             model_onnx = onnx.load(f)
 
-        call_tflite(imgs[0])
+        interpreter_details = tf.lite.Interpreter(tname, experimental_preserve_all_tensors=True)
+        input_details = interpreter_details.get_input_details()
+        index_in = input_details[0]['index']
+        interpreter_details.allocate_tensors()
+        interpreter_details.set_tensor(index_in, imgs[0])
+        interpreter_details.invoke()
+        details = interpreter_details.get_tensor_details()
+
         inputs = {input_name: imgs[0]}
-        details = interpreter.get_tensor_details()
         names_index = {}
         for tt in details:
             names_index[tt['name']] = (tt['index'], tt['quantization'], tt['quantization_parameters'])
@@ -414,7 +420,7 @@ def call_tflite(inp):
         for name_tfl, name_ort in names:
             index = names_index[name_tfl]
 
-            tfl_value = interpreter.get_tensor(index[0])
+            tfl_value = interpreter_details.get_tensor(index[0])
 
             new_name = onnx_name + ".%s.onnx" % name_ort.replace(":", "_").replace(";", "_").replace("/", "_")
             if not os.path.exists(new_name):