B
    )`
                 @   s   d Z ddlmZ ddlmZ ddlm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re 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 ) z5Offline dump analyzer of TensorFlow Debugger (tfdbg).    )absolute_import)division)print_functionN)analyzer_cli)
debug_data)appc             C   sz   t jr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)Ztensor_filtersui_typeztfdbg offline @ %sZblack_on_whitelt)titleZtitle_colorZinit_command)FLAGSZ	log_usageZdump_dirprintsysstderrexitr   ZDebugDumpDirZvalidate_graphr   Zcreate_analyzer_uir   r   Zrun_ui)_Z
debug_dumpclir    r   [/home/dcms/DCMS/lib/python3.7/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>5       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_typeZcursesz4Command-line user interface type (curses | readline)z--validate_graphzN      Whether the dumped tensors will be validated against the GraphDefs      )r%   r&   r   r"   r#   )r   argv)__doc__
__future__r   r   r   argparser   Ztensorflow.python.debug.clir   Ztensorflow.python.debug.libr   Ztensorflow.python.platformr   r   __name__ArgumentParserparserregisteradd_argumentstrparse_known_argsr   Zunparsedrunr'   r   r   r   r   <module>   sF   