o
    i                      @   s|   d Z ddlZddlmZmZ ejed ej	dds%ddl
mZ nddlmZ G dd dejZG d	d
 d
eZdS )zPython TFLite metrics helper.    N)OptionalTexttflite_runtimemetrics_portable)metrics_interfacec                   @   sl   e Zd ZdZ		ddee dee ddfddZdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd ZdS )TFLiteMetricszTFLite metrics helper.N
model_hash
model_pathreturnc                 C      d S N )selfr   r	   r   r   \/mnt/sdb/aimis/docanh/lib/python3.10/site-packages/tensorflow/lite/python/metrics/metrics.py__init__!   s   zTFLiteMetrics.__init__c                 C   r   r   r   r   r   r   r   "increase_counter_debugger_creation&      z0TFLiteMetrics.increase_counter_debugger_creationc                 C   r   r   r   r   r   r   r   %increase_counter_interpreter_creation)   r   z3TFLiteMetrics.increase_counter_interpreter_creationc                 C   r   r   r   r   r   r   r   "increase_counter_converter_attempt,   r   z0TFLiteMetrics.increase_counter_converter_attemptc                 C   r   r   r   r   r   r   r   "increase_counter_converter_success/   r   z0TFLiteMetrics.increase_counter_converter_successc                 C   r   r   r   )r   namevaluer   r   r   set_converter_param2   r   z!TFLiteMetrics.set_converter_paramc                 C   r   r   r   )r   
error_datar   r   r   set_converter_error5   r   z!TFLiteMetrics.set_converter_errorc                 C   r   r   r   )r   r   r   r   r   set_converter_latency8   r   z#TFLiteMetrics.set_converter_latency)NN)__name__
__module____qualname____doc__r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s$    
r   c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	TFLiteConverterMetricsz7Similar to TFLiteMetrics but specialized for converter.c                 C   r   r   r   r   r   r   r   __del__?   r   zTFLiteConverterMetrics.__del__c                 C   r   r   r   r   r   r   r   set_export_requiredB   r   z*TFLiteConverterMetrics.set_export_requiredc                 C   r   r   r   r   r   r   r   export_metricsE   r   z%TFLiteConverterMetrics.export_metricsN)r   r   r   r    r"   r#   r$   r   r   r   r   r!   <   s
    r!   )r    ostypingr   r   pathsplitext__file__endswithjointensorflow.lite.python.metricsr   r   TFLiteMetricsInterfacer   r!   r   r   r   r   <module>   s   