tf.lite.experimental.Analyzer

Предоставляет набор инструментов анализатора моделей TFLite.

Пример:

model = tf.keras.applications.MobileNetV3Large()
fb_model = tf.lite.TFLiteConverterV2.from_keras_model(model).convert()
tf.lite.experimental.Analyzer.analyze(model_content=fb_model)
# === TFLite ModelAnalyzer ===
#
# Your TFLite model has ‘1’ subgraph(s). In the subgraph description below,
# T# represents the Tensor numbers. For example, in Subgraph#0, the MUL op
# takes tensor #0 and tensor #19 as input and produces tensor #136 as output.
#
# Subgraph#0 main(T#0) -> [T#263]
#   Op#0 MUL(T#0, T#19) -> [T#136]
#   Op#1 ADD(T#136, T#18) -> [T#137]
#   Op#2 CONV_2D(T#137, T#44, T#93) -> [T#138]
#   Op#3 HARD_SWISH(T#138) -> [T#139]
#   Op#4 DEPTHWISE_CONV_2D(T#139, T#94, T#24) -> [T#140]
#   ...

Методы

analyze

Просмотреть исходный код

@staticmethod
analyze(
    model_path=None, model_content=None, gpu_compatibility=False, **kwargs
)

Анализирует заданную модель tflite_model, выгружая структуру модели.

Этот инструмент позволяет понять модель пользователя TFLite flatbuffer, выгружая внутреннюю структуру графа. Он также предоставляет дополнительные функции, такие как проверка совместимости делегата GPU.