B
    ð»ˆd  ã               @   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   ZTocoConvert)Zmodel_flags_strZtoco_flags_strÚinput_data_strZdebug_info_strZenable_mlir_converter© r   úR/var/www/html/venv/lib/python3.7/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   ZExperimentalMlirQuantizeModel)
r   Zdisable_per_channelZfully_quantizeZinference_typeZinput_data_typeZoutput_data_typeZenable_numeric_verifyZenable_whole_model_verifyZdenylisted_opsZdenylisted_nodesr   r   r   Ú"wrapped_experimental_mlir_quantize$   s    r   c             C   s
   t  | ¡S )z'Wraps experimental mlir sparsify model.)r   ZExperimentalMlirSparsifyModel)r   r   r   r   Ú"wrapped_experimental_mlir_sparsify1   s    r	   c             C   s
   t  | ¡S )z,Wraps RegisterCustomOpdefs with lazy loader.)r   ZRegisterCustomOpdefs)Zcustom_opdefs_listr   r   r   Úwrapped_register_custom_opdefs6   s    r
   c               C   s   t  ¡ S )z/Wraps RetrieveCollectedErrors with lazy loader.)r   ZRetrieveCollectedErrorsr   r   r   r   Ú!wrapped_retrieve_collected_errors;   s    r   c             C   s   t  | |¡S )z,Wraps FlatBufferFileToMlir with lazy loader.)r   ZFlatBufferToMlir)ÚmodelZinput_is_filepathr   r   r   Ú wrapped_flat_buffer_file_to_mlir@   s    r   N)
Ú__doc__Ztensorflow.pythonr   r   r   r   r	   r
   r   r   r   r   r   r   Ú<module>   s   