a
    -SicÑ  ã                   @   sb   d Z ddlmZ ddlmZ ddlT dddg fdd„Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
dd„ ZdS )z6Python module for MLIR functions exported by pybind11.é    )Úpywrap_tensorflow)Úcontext)Ú*Nc              
   C   sx   |d ur\t t| ƒ d¡| d¡|d |¡ d¡d |¡ d¡d |¡ d¡d |¡ d¡ƒS t t| ƒ d¡| d¡|ƒS )Núutf-8ú,ú:)ZImportGraphDefÚstrÚencodeÚjoin)ÚgraphdefÚpass_pipelineÚshow_debug_infoÚinput_namesZinput_data_typesZinput_data_shapesÚoutput_names© r   úY/var/www/html/django/DPS/env/lib/python3.9/site-packages/tensorflow/python/pywrap_mlir.pyÚimport_graphdef   s    ûþr   c                 C   s2   t   ¡ }| ¡  t|jt| jƒ d¡| d¡|ƒS ©Nr   )r   Úensure_initializedZImportFunctionÚ_handler   Úfunction_defr	   )Úconcrete_functionr   r   Úctxtr   r   r   Úimport_function*   s    
þr   c                 C   s    t t| ƒ d¡t|ƒ d¡|ƒS r   )Z#ExperimentalConvertSavedModelToMlirr   r	   )Úsaved_model_pathÚexported_namesr   r   r   r   Ú(experimental_convert_saved_model_to_mlir2   s    þr   c                 C   s.   t t| ƒ d¡t|ƒ d¡t|ƒ d¡||ƒS r   )Z)ExperimentalConvertSavedModelV1ToMlirLiter   r	   )r   r   ÚtagsÚupgrade_legacyr   r   r   r   Ú0experimental_convert_saved_model_v1_to_mlir_lite9   s
    ýr   c                 C   s0   t t| ƒ d¡t|ƒ d¡t|ƒ d¡|||ƒS r   )Z%ExperimentalConvertSavedModelV1ToMlirr   r	   )r   r   r   Zlift_variablesr   r   r   r   r   Ú+experimental_convert_saved_model_v1_to_mlirC   s    ür    c                 C   s   t |  d¡| d¡|ƒS r   )ZExperimentalRunPassPipeliner	   )Zmlir_txtr   r   r   r   r   Úexperimental_run_pass_pipelineN   s    ÿr!   )Ú__doc__Útensorflow.pythonr   Útensorflow.python.eagerr   Ztensorflow.python._pywrap_mlirr   r   r   r   r    r!   r   r   r   r   Ú<module>   s   ú

