o
    i e0	                     @   s   d Z ddlZddlZddlmZ ddlmZ ddlmZ dd Z	e
dkrne Zed	d
dd  ejdeddd ejdd
ddddd ejdeddd ejdddd
ddd e \ZZeje	ejd ge d dS dS )z5Offline dump analyzer of TensorFlow Debugger (tfdbg).    N)app)analyzer_cli)
debug_datac                 C   st   t jstdtjd td tdt j  tjt jt jd}t	j
|dtjit jd}dt j }|j|d	d
d d S )NzERROR: dump_dir flag is empty.)file   z"tfdbg offline: FLAGS.dump_dir = %s)validatehas_inf_or_nan)tensor_filtersui_typeztfdbg offline @ %sblack_on_whitelt)titletitle_colorinit_command)FLAGSdump_dirprintsysstderrexitr   DebugDumpDirvalidate_graphr   create_analyzer_uir   r
   run_ui)_
debug_dumpclir    r   [/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/debug/cli/offline_analyzer.pymain   s   

r   __main__typeboolc                 C   s   |   dkS )Ntrue)lower)vr   r   r   <lambda>-   s    r&   z
--dump_dir ztfdbg dump directory to load)r!   defaulthelpz--log_usage?Tz.Whether the usage of this tool is to be logged)r!   nargsconstr(   r)   z	--ui_typereadlinez=Command-line user interface type (only readline is supported)z--validate_graphzN      Whether the dumped tensors will be validated against the GraphDefs      )r+   r,   r!   r(   r)   )r   argv)__doc__argparser   abslr   tensorflow.python.debug.clir   tensorflow.python.debug.libr   r   __name__ArgumentParserparserregisteradd_argumentstrparse_known_argsr   unparsedrunr.   r   r   r   r   <module>   sL   	