o
    ®Ÿ?eL  ã                   @   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ú,ú:)ZImportGraphDefÚstrÚencodeÚjoin)ZgraphdefÚpass_pipelineÚshow_debug_infoZinput_namesZinput_data_typesZinput_data_shapesZoutput_names© r   ú^/home/www/facesmatcher.com/pyenv/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   Zensure_initializedZImportFunctionZ_handler   Zfunction_defr	   )Zconcrete_functionr   r   Zctxtr   r   r   Úimport_function1   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_mlir<   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_liteF   s   ûr   c                 C   s2   t t| ƒ d¡t|ƒ d¡t|ƒ d¡||||ƒS r   )Z%ExperimentalConvertSavedModelV1ToMlirr   r	   )r   r   r   Zlift_variablesZ!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   )ZExperimentalRunPassPipeliner	   )Úmlir_txtr   r   r   r   r   Úexperimental_run_pass_pipelinef   s   ÿr   c                 C   s   t |  d¡| ¡ ƒS r   )Z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   )Z ExperimentalTFLiteToTosaBytecoder	   )Z
flatbufferÚbytecodeZuse_external_constantZordered_input_arraysZordered_output_arraysr   r   r   Ú$experimental_tflite_to_tosa_bytecodep   s   ûr   )FNN)Ú__doc__Ztensorflow.pythonr   Ztensorflow.python.eagerr   Ztensorflow.python._pywrap_mlirr   r   r   r   r   r   r   r   r   r   r   r   Ú<module>   s&   
ù
û