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.
| Аргументы | |
|---|---|
model_path | Путь к модели TFLite flatbuffer. |
model_content | Объект модели TFLite flatbuffer. |
gpu_compatibility | Необходимо ли проверить совместимость с делегатом GPU. |
**kwargs | Экспериментальные ключевые аргументы для анализа API. |
| Возвращаемые значения | |
|---|---|
| Вывод аналитического отчета через консоль. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/lite/experimental/Analyzer