o
    Ô2úhL  ã                   @   sz   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d„ Z			d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ú,ú:)ÚImportGraphDefÚstrÚencodeÚjoin)ÚgraphdefÚpass_pipelineÚshow_debug_infoÚinput_namesÚinput_data_typesÚinput_data_shapesÚoutput_names© r   úX/var/www/html/chatgem/venv/lib/python3.10/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_initializedÚImportFunctionÚ_handler	   Úfunction_defr
   )Úconcrete_functionr   r   Úctxtr   r   r   Úimport_function1   s   ür   c                 C   s    t t| ƒ d¡t|ƒ d¡|ƒS r   )Ú#ExperimentalConvertSavedModelToMlirr	   r
   )Úsaved_model_pathÚexported_namesr   r   r   r   Ú(experimental_convert_saved_model_to_mlir<   s
   ýr!   c                 C   s.   t t| ƒ d¡t|ƒ d¡t|ƒ d¡||ƒS r   )Ú)ExperimentalConvertSavedModelV1ToMlirLiter	   r
   )r   r    ÚtagsÚupgrade_legacyr   r   r   r   Ú0experimental_convert_saved_model_v1_to_mlir_liteF   s   ûr%   c                 C   s2   t t| ƒ d¡t|ƒ d¡t|ƒ d¡||||ƒS r   )Ú%ExperimentalConvertSavedModelV1ToMlirr	   r
   )r   r    r#   Úlift_variablesÚ!include_variables_in_initializersr$   r   r   r   r   Ú+experimental_convert_saved_model_v1_to_mlirR   s   	ùr)   c                 C   s   t |  d¡| d¡|ƒS r   )ÚExperimentalRunPassPipeliner
   )Úmlir_txtr   r   r   r   r   Úexperimental_run_pass_pipelinef   s   ÿr,   c                 C   s   t |  d¡| ¡ ƒS r   )ÚExperimentalWriteBytecoder
   )Úfilenamer+   r   r   r   Úexperimental_write_bytecodel   s   r/   Fc                 C   s4   |d u rg }|d u rg }t |  d¡| d¡|||ƒS r   )Ú ExperimentalTFLiteToTosaBytecoder
   )Ú
flatbufferÚbytecodeÚuse_external_constantÚordered_input_arraysÚordered_output_arraysr   r   r   Ú$experimental_tflite_to_tosa_bytecodep   s   ûr6   )FNN)Ú__doc__Útensorflow.pythonr   Útensorflow.python.eagerr   Útensorflow.python._pywrap_mlirr   r   r!   r%   r)   r,   r/   r6   r   r   r   r   Ú<module>   s&   
ù
û