tflite_output_layer.py

import tensorflow as tf


saved_model_dir = "/Users/cuiqingli123/Workspace/torch_op_exp/mobile_net_v3_small"
# Convert the model
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir) # path to the SavedModel directory
converter.optimizations = [tf.lite.Optimize.DEFAULT]
tflite_model = converter.convert()


interpreter = tf.lite.Interpreter(model_content=tflite_model)
all_tensors = interpreter.get_tensor_details()

for tensor_layer_info in all_tensors:
    print(tensor_layer_info)