o
    ·i e¯
  ã                   @   sP   d Z ddlmZ ddlmZ dd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Zdd„ Z	dS )z-Wraps toco interface with python lazy loader.é    )Úpywrap_tensorflow)Ú_pywrap_toco_apic                 C   s   t  | ||d||¡S )z#Wraps TocoConvert with lazy loader.F)r   ÚTocoConvert)Úmodel_flags_strÚtoco_flags_strÚinput_data_strÚdebug_info_strÚenable_mlir_converter© r
   úO/var/www/myenv/lib/python3.10/site-packages/tensorflow/lite/python/wrap_toco.pyÚwrapped_toco_convert   s   úr   c                 C   s   t  | |||||||||	|
¡S )z'Wraps experimental mlir quantize model.)r   ÚExperimentalMlirQuantizeModel)r   Údisable_per_channelÚfully_quantizeÚinference_typeÚinput_data_typeÚoutput_data_typeÚenable_numeric_verifyÚenable_whole_model_verifyÚdenylisted_opsÚdenylisted_nodesÚenable_variable_quantizationr
   r
   r   Ú"wrapped_experimental_mlir_quantize$   s   ür   c                 C   ó
   t  | ¡S )z'Wraps experimental mlir sparsify model.)r   ÚExperimentalMlirSparsifyModel)r   r
   r
   r   Ú"wrapped_experimental_mlir_sparsify1   ó   
r   c                 C   r   )z,Wraps RegisterCustomOpdefs with lazy loader.)r   ÚRegisterCustomOpdefs)Úcustom_opdefs_listr
   r
   r   Úwrapped_register_custom_opdefs6   r   r   c                   C   s   t  ¡ S )z/Wraps RetrieveCollectedErrors with lazy loader.)r   ÚRetrieveCollectedErrorsr
   r
   r
   r   Ú!wrapped_retrieve_collected_errors;   s   r!   c                 C   s   t  | |¡S )z,Wraps FlatBufferFileToMlir with lazy loader.)r   ÚFlatBufferToMlir)ÚmodelÚinput_is_filepathr
   r
   r   Ú wrapped_flat_buffer_file_to_mlir@   s   r%   N)
Ú__doc__Útensorflow.pythonr   r   r   r   r   r   r!   r%   r
   r
   r
   r   Ú<module>   s   