BVhedZddlZejjedj ejj ddsddlmZddl m Z ddl m Zn ddlm Z dZed Gd d Zy) z+This tool analyzes a TensorFlow Lite graph.Ntflite_runtimeanalyzer)wrap_converter)_pywrap_analyzer_wrapper) tf_exportc ~~dS)Nc|S)N)xs O/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/lite/python/analyzer.pyz_tf_export.. sQr )r kwargss r _tf_exportrs  6 rzlite.experimental.Analyzerc(eZdZdZe ddZy) ModelAnalyzerupProvides a collection of TFLite model analyzer tools. Example: ```python 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] # ... ``` WARNING: Experimental interface, subject to change. Nc |s |s td|rtd|d|}d}ntd|}d}|jddr ttj||ytt j |||y) aAnalyzes the given tflite_model with dumping model structure. This tool provides a way to understand users' TFLite flatbuffer model by dumping internal graph structure. It also provides additional features like checking GPU delegate compatibility. WARNING: Experimental interface, subject to change. The output format is not guaranteed to stay stable, so don't write scripts to this. Args: model_path: TFLite flatbuffer model path. model_content: TFLite flatbuffer model object. gpu_compatibility: Whether to check GPU delegate compatibility. **kwargs: Experimental keyword arguments to analyze API. Returns: Print analyzed report via console output. z4neither `model_path` nor `model_content` is providedz=== z === Tz=== TFLite ModelAnalyzer === Fexperimental_use_mlirN) ValueErrorprintgetr wrapped_flat_buffer_file_to_mlir_analyzer_wrapperr) model_path model_contentgpu_compatibilityr tflite_modelinput_is_filepaths r analyzezModelAnalyzer.analyze?s0 m M NN d:,f %&l ,-"l zz)51  9 9-    ) ),8I*; =>r)NNF)__name__ __module__ __qualname____doc__ staticmethodrr rr rr#s$4 %+>+>rr)r#ospathsplitext__file__endswithjoin$tensorflow.compiler.mlir.lite.pythonr'tensorflow.lite.python.analyzer_wrapperrr tensorflow.python.util.tf_exportrrrrr rr r.s}2  ww!!$--GGLL!:.0BcFK  ()G>G>*G>r