B )²ô`Fåã@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z dd l m Z dd l mZdd l mZdd l mZdd lmZddlmZejZdZdZdZdZdZdZdZdZdZdZdZ d"dd„Z!Gdd„de"ƒZ#d#d d!„Z$dS)$z¾CLI Backend for the Analyzer Part of the Debugger. The analyzer performs post hoc analysis of dumped intermediate tensors and graph structure information from debugged Session.run() calls. é)Úabsolute_import)Údivision)Úprint_functionN)Úxrange)Ú cli_config)Ú cli_shared)Úcommand_parser)Údebugger_cli_common)Ú evaluator)Ú ui_factory)Ú debug_graphs)Ú source_utilsz| z z|- z(%d) z[%s] z(Ctrl) z...Ú timestampZ dump_sizeÚop_typeÚ tensor_nameTcCst ¡}| tjdd|d¡|rŒ| tjdd||d¡| tjdd||d¡| tjdd||d¡| tjd d ||d¡nX| tjdd d d¡| tjdd d d¡| tjdd d d¡| tjd d d d¡| t d d ¡¡| t dd¡¡||jtj<d S)a2Generate main menu for the screen output from a command. Args: output: (debugger_cli_common.RichTextLines) the output object to modify. node_name: (str or None) name of the node involved (if any). If None, the menu items node_info, list_inputs and list_outputs will be automatically disabled, overriding the values of arguments enable_node_info, enable_list_inputs and enable_list_outputs. enable_list_tensors: (bool) whether the list_tensor menu item will be enabled. enable_node_info: (bool) whether the node_info item will be enabled. enable_print_tensor: (bool) whether the print_tensor item will be enabled. enable_list_inputs: (bool) whether the item list_inputs will be enabled. enable_list_outputs: (bool) whether the item list_outputs will be enabled. Ú list_tensors)ZenabledÚ node_infoznode_info -a -d -t %sÚ print_tensorzprint_tensor %sÚ list_inputszlist_inputs -c -r %sÚ list_outputszlist_outputs -c -r %sNFZrun_infoÚhelp)r ZMenuÚappendÚMenuItemÚ annotationsZ MAIN_MENU_KEY)ÚoutputÚ node_nameÚenable_list_tensorsÚenable_node_infoÚenable_print_tensorÚenable_list_inputsÚenable_list_outputsÚmenu©r"úW/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/python/debug/cli/analyzer_cli.pyÚ_add_main_menu=sN    r$c@seZdZdZdZdZdZdZdZdd„Z d d „Z d d „Z d d„Z dd„Z dd d!„Zd?d"d#„Zd@d$d%„ZdAd&d'„ZdBd)d*„ZdCd+d,„Zd-d.„ZdDd/d0„ZdEd2d3„ZdFd4d5„Zd6d7„Zd8d9„Z d:d;„Z!dS)GÚ DebugAnalyzerz.Analyzer for debug data from dump directories.zt (ms)zSize (B)zOp typez Tensor name)Z_SendZ_RecvZ _HostSendZ _HostRecvZ_RetvalcCs6||_t |j¡|_i|_| |¡| d|j¡dS)z²DebugAnalyzer constructor. Args: debug_dump: A DebugDumpDir object. config: A `cli_config.CLIConfig` object that carries user-facing configurations. Úgraph_recursion_depthN)Ú _debug_dumpr ZExpressionEvaluatorÚ _evaluatorÚ_tensor_filtersÚ_build_argument_parsersZ set_callback)ÚselfÚ debug_dumpÚconfigr"r"r#Ú__init__Žs  zDebugAnalyzer.__init__c Csi|_tjdtjd}|jdddtddd|jd d d tdd d|jd ddtddd|jdddtddd|jdddttdtttt fd|jdddddd||jd<tjd tjd}|jd!td"d#|jd$d%d&dd'd|jd(d)d*dd+d|jdd,d-dd.d||jd/<tjd0tjd}|jd!td1d#|jd2d3dd4d5|jd(d6d7t |  d8¡d9d|jdd:d;ddd5||jd?<tjd@tjd}|jd!td1d#|jd2d3dd4d5|jd(d6d7t |  d8¡dAd|jdd:d;ddBd|jdd=ddCd5||jdD<t   dE¡|jdF<tjdGtjd}|jdHtdId#|jddJdKddLd|jdMdNt dOdPdQ|jdRdSt dTdUdQ||jdV<tjdWtjd}|jdXdYtddZdQ|jd dtdd[dQ||jd\<tjd]tjd}|jd^td_d#|jd$d`daddbd|jdcddddedf||jdg<dhS)iz¿Build argument parsers for DebugAnalayzer. Args: config: A `cli_config.CLIConfig` object. Returns: A dict mapping command handler name to `ArgumentParser` instance. z!List dumped intermediate tensors.)Ú descriptionÚusagez-fz--tensor_filterÚ tensor_filterÚz:List only Tensors passing the filter of the specified name)ÚdestÚtypeÚdefaultrz-fennz--filter_exclude_node_namesÚfilter_exclude_node_namesz‹When applying the tensor filter, exclude node with names matching the regular expression. Applicable only if --tensor_filter or -f is used.z-nz--node_name_filterÚnode_name_filterzfilter node name by regex.z-tz--op_type_filterÚop_type_filterzfilter op type by regex.z-sz --sort_byÚsort_byz2the field to sort the data by: (%s | %s | %s | %s)z-rz --reverseÚreverseÚ store_truez+sort the data in reverse (descending) order)r3ÚactionrrzShow information about a node.rzdName of the node or an associated tensor, e.g., hidden1/Wx_plus_b/MatMul, hidden1/Wx_plus_b/MatMul:0)r4rz-az --attributesÚ attributesz!Also list attributes of the node.z-dz--dumpsÚdumpsz(Also list dumps available from the node.z --tracebackÚ tracebackzKAlso include the traceback of the node's creation (if available in Python).rzShow inputs to a node.znName of the node or an output tensor from the node, e.g., hidden1/Wx_plus_b/MatMul, hidden1/Wx_plus_b/MatMul:0z-cz --controlzInclude control inputs.)r<rz--depthÚdepthr&z:, e.g., hidden_0/MatMul:0. 2) if the default debug op "DebugIdentity" is to be overridden, use ::, e.g., hidden_0/MatMul:0:DebugNumericSummary. 3) if the tensor of the same name exists on more than one device, use ::[:], e.g., /job:worker/replica:0/task:0/gpu:0:hidden_0/MatMul:0 /job:worker/replica:0/task:2/cpu:0:hidden_0/MatMul:0:DebugNanCount. 4) if the tensor is executed multiple times in a given `Session.run` call, specify the execution index with a 0-based integer enclose in a pair of brackets at the end, e.g., RNN/tanh:0[0] /job:worker/replica:0/task:0/gpu:0:RNN/tanh:0[0].z--allÚ print_allz\Print the tensor in its entirety, i.e., do not use ellipses (may be slow for large results).z-wz --write_pathzLPath of the numpy file to write the evaluation result to, using numpy.save())r5rÚevalN)Ú _arg_parsersÚargparseÚArgumentParserÚSUPPRESSÚ add_argumentÚstrÚSORT_TENSORS_BY_TIMESTAMPÚSORT_TENSORS_BY_DUMP_SIZEÚSORT_TENSORS_BY_OP_TYPEÚSORT_TENSORS_BY_TENSOR_NAMEÚintÚgetrZget_print_tensor_argparser)r+r-Zapr"r"r#r*£s~                z%DebugAnalyzer._build_argument_parserscCs<t|tƒstdƒ‚|stdƒ‚t|ƒs.tdƒ‚||j|<dS)a@Add a tensor filter. A tensor filter is a named callable of the signature: filter_callable(dump_datum, tensor), wherein dump_datum is an instance of debug_data.DebugTensorDatum carrying metadata about the dumped tensor, including tensor name, timestamps, etc. tensor is the value of the dumped tensor as an numpy.ndarray object. The return value of the function is a bool. This is the same signature as the input argument to debug_data.DebugDumpDir.find(). Args: filter_name: (str) name of the filter. Cannot be empty. filter_callable: (callable) a filter function of the signature described as above. Raises: ValueError: If filter_name is an empty str. TypeError: If filter_name is not a str. Or if filter_callable is not callable. z=Input argument filter_name is expected to be str, but is not.z+Input argument filter_name cannot be empty.zFInput argument filter_callable is expected to be callable, but is not.N)Ú isinstancerPÚ TypeErrorÚ ValueErrorÚcallabler))r+Ú filter_nameÚfilter_callabler"r"r#Úadd_tensor_filter¤s zDebugAnalyzer.add_tensor_filtercCs ||jkrtd|ƒ‚|j|S)aRetrieve filter function by name. Args: filter_name: Name of the filter set during add_tensor_filter() call. Returns: The callable associated with the filter name. Raises: ValueError: If there is no tensor filter of the specified filter name. z$There is no tensor filter named "%s")r)rY)r+r[r"r"r#Úget_tensor_filterÌs  zDebugAnalyzer.get_tensor_filtercCs|j| ¡S)N)rKÚ format_help)r+Z handler_namer"r"r#Úget_helpÞszDebugAnalyzer.get_helpNc Cs |}|jd |¡}g}g}|jr@t |j¡}| d|j¡nd}|jrht |j¡}| d|j¡nd}t |¡}| d¡|j rÞy|  |j ¡} Wn2t k rÈt   d|j ¡}t|ddd|SX|jj| |jd } n|jrìt d ƒ‚|jj} | | ¡\} } } | | |j|j¡} | | || | | ¡¡d }x*| D] }|rV| |j¡sVq8|rz|j |j¡}| |¡szq8|j|jjd }t  |j¡}d |j|j f}|j |j¡}d|}|d| t!|ƒ7}||7}|d| | t!|ƒ7}||7}|d| | | t!|ƒ7}||7}|j|t!|ƒt!|ƒt!|ƒt "dd|¡fgd|d7}q8W|j r~| #d||j fg¡n| #d|g¡t|ddd|S)aäCommand handler for list_tensors. List tensors dumped during debugged Session.run() call. Args: args: Command-line arguments, excluding the command prefix, as a list of str. screen_info: Optional dict input containing screen information such as cols. Returns: Output text lines as a RichTextLines object. Raises: ValueError: If `--filter_exclude_node_names` is used without `-f` or `--tensor_filter` being used. rzOp type regex filter: "%s"NzNode name regex filter: "%s"r2z%There is no tensor filter named "%s".F)rr)Zexclude_node_namesz_The flag --filter_exclude_node_names is valid only when the flag -f or --tensor_filter is used.rg@@z%s:%dz[%.3f]ú zpt %s)Úfont_attr_segsrEz(%d dumped tensor(s) passing filter "%s":z%d dumped tensor(s):)$rKÚ parse_argsr8ÚreÚcompilerr7r Ú RichTextLinesr1r^rYrÚerrorr$r'Úfindr6Údumped_tensor_dataÚ"_measure_tensor_list_column_widthsÚ_sort_dump_data_byr9r:ÚextendÚ_tensor_list_column_headsÚmatchrÚ node_op_typerÚt0Úbytes_to_readable_strÚdump_size_bytesÚ output_slotÚlenrÚprepend)r+ÚargsÚ screen_infoÚ_ÚparsedrZ filter_strsZ op_type_regexZnode_name_regexr\Z data_to_showÚmax_timestamp_widthÚmax_dump_size_widthÚmax_op_type_widthÚ dump_countÚdumprÚrel_timeÚ dump_size_strZdumped_tensor_nameÚliner"r"r#rásˆ          zDebugAnalyzer.list_tensorsc Csêd}|r.|dj|jjd}td|ƒd}t|t|jƒdƒ}d}x4|D],}t |j¡}t|ƒd|krLt|ƒd}qLWt|t|j ƒdƒ}d}x6|D].}|j  |j ¡}t|ƒd|kršt|ƒd}qšWt|t|j ƒdƒ}|||fS)a Determine the maximum widths of the timestamp and op-type column. This method assumes that data is sorted in the default order, i.e., by ascending timestamps. Args: data: (list of DebugTensorDaum) the data based on which the maximum column widths will be determined. Returns: (int) maximum width of the timestamp column. 0 if data is empty. (int) maximum width of the dump size column. 0 if data is empty. (int) maximum width of the op type column. 0 if data is empty. réÿÿÿÿg@@z[%.3f] rE) rr'rprtÚmaxÚ_TIMESTAMP_COLUMN_HEADrrqrrÚ_DUMP_SIZE_COLUMN_HEADrorÚ_OP_TYPE_COLUMN_HEAD) r+ÚdatarzZmax_rel_time_msr{r~r€r|rr"r"r#rjXs*   z0DebugAnalyzer._measure_tensor_list_column_widthscs||tkrt||dd„dS|tkr4t||dd„dS|tkrRt||‡fdd„dS|tkrlt||dd„dStd|ƒ‚dS) aoSort a list of DebugTensorDatum in specified order. Args: data: (list of DebugTensorDatum) the data to be sorted. sort_by: The field to sort data by. reverse: (bool) Whether to use reversed (descending) order. Returns: (list of DebugTensorDatum) in sorted order. Raises: ValueError: given an invalid value of sort_by. cSs|jS)N)r)Úxr"r"r#Ú”óz2DebugAnalyzer._sort_dump_data_by..)r:ÚkeycSs|jS)N)rr)rˆr"r"r#r‰–rŠcsˆj |j¡S)N)r'ror)rˆ)r+r"r#r‰›rŠcSsd|j|jfS)Nz%s:%d)rrs)rˆr"r"r#r‰ rŠz&Unsupported key to sort tensors by: %sN)rQÚsortedrRrSrTrY)r+r‡r9r:r")r+r#rks$  z DebugAnalyzer._sort_dump_data_byc Cs d}|jr|d|j7}|jr,|d|j7}|jr@|d|j7}dgi}|j}d|tf}|jtkrr|jsr|d7}|d dt|ƒt   d|¡d gf¡|d |t|ƒ7}t|ƒ} ||j 7}d|t f}|jt krà|jsà|d7}|d | t|ƒt   d|¡d gf¡|d ||t|ƒ7}t|ƒ} ||j 7}d|tf}|jtkrV|jsV|d7}|d | t|ƒt   d|¡d gf¡|d |||t|ƒ7}t|ƒ} ||j7}d|tf}|jtkrÐ|jsÐ|d7}|d | t|ƒt   d |¡d gf¡|d |||t|ƒ7}t j|g|d S) a™Generate a line containing the column heads of the tensor list. Args: parsed: Parsed arguments (by argparse) of the list_tensors command. max_timestamp_width: (int) maximum width of the timestamp column. max_dump_size_width: (int) maximum width of the dump size column. max_op_type_width: (int) maximum width of the op type column. Returns: A RichTextLines object. rz -f %sz -t %sz -n %srz%s -s %sz -rNÚboldrar2)rb)r1r8r7r„rQr9r:rrtr rr…rRr†rSÚ_TENSOR_NAME_COLUMN_HEADrTrf) r+ryrzr{r|Ú base_commandÚ attr_segsÚrowÚcommandZprev_lenr"r"r#rm¤sT       z'DebugAnalyzer._tensor_list_column_headscCsž|}|jd |¡}t |j¡\}}|j |¡sVt d|¡}t |dddddd|Sd|g}dt |d ƒt |ƒt |d ƒd fgi} |  d ¡|  d |j  |¡¡|  d |j  |¡¡tj|| d}| |j |¡¡} | |jj|dd¡} | | d| | ¡¡| |j |¡¡} | |jj|dd¡} | | d| | ¡¡|jr\| | |¡¡|jrt| | |¡¡|jrŒ| | |¡¡t ||dd|S)aQCommand handler for node_info. Query information about a given node. Args: args: Command-line arguments, excluding the command prefix, as a list of str. screen_info: Optional dict input containing screen information such as cols. Returns: Output text lines as a RichTextLines object. rz3There is no node named "%s" in the partition graphsNTF)rrrrr zNode %srr‚rr2z Op: %sz Device: %s)rb)Ú is_controlÚinputZ recipient)rr)rKrcr Úparse_node_or_tensor_namerr'Ú node_existsrrgr$rtrroZ node_devicer rfÚ_exclude_blacklisted_opsÚ node_inputsrlÚ_format_neighborsÚnode_recipientsr=Ú_list_node_attributesr>Ú_list_node_dumpsr?Ú_render_node_traceback)r+rvrwrxryrZ unused_slotrÚlinesrbÚinputsÚ ctrl_inputsZrecsZ ctrl_recsr"r"r#räsP    (  zDebugAnalyzer.node_infocs‡fdd„|DƒS)zÙExclude all nodes whose op types are in _GRAPH_STRUCT_OP_TYPE_BLACKLIST. Args: node_names: An iterable of node or graph element names. Returns: A list of node names that are not blacklisted. cs(g|] }ˆj t |¡¡ˆjkr|‘qSr")r'ror Ú get_node_nameÚ_GRAPH_STRUCT_OP_TYPE_BLACKLIST)Ú.0r)r+r"r#ú Bs z:DebugAnalyzer._exclude_blacklisted_ops..r")r+Z node_namesr")r+r#r—9s z&DebugAnalyzer._exclude_blacklisted_opsc Cstdƒtdƒtddƒg}y²|j |¡}x t|ƒD]”\}\}}}}| d||f¡|rjt dd||f¡nd} tdƒ} | td|| ƒ7} | | ¡| d |¡| d |r²d |nd ¡| d¡q0WWn<tk rè| d ¡Yn tk r| d¡YnXt  |¡S)zßRender traceback of a node's creation in Python, if available. Args: node_name: (str) name of the node. Returns: A RichTextLines object containing the stack trace of the node's construction. r2zTraceback of node construction:rz%d: %sz ps %s -b %dNz z Line: %dz Function: %sz Text: z"%s"ÚNonez-(Node unavailable in the loaded Python graph)z5(Unavailable because no Python graph has been loaded)) ÚRLr'Znode_tracebackÚ enumeraterr rÚKeyErrorÚ LookupErrorÚ#rich_text_lines_from_rich_line_list) r+rržZ node_stackr@Ú file_pathrZ function_nameÚtextÚ attributeZline_number_liner"r"r#rGs$   z$DebugAnalyzer._render_node_tracebackcCsR|}|jd |¡}|j|j|j|j|j|jdd}t  |j¡}t ||dd|S)aECommand handler for inputs. Show inputs to a given node. Args: args: Command-line arguments, excluding the command prefix, as a list of str. screen_info: Optional dict input containing screen information such as cols. Returns: Output text lines as a RichTextLines object. rF)Ú do_outputs)rr) rKrcÚ_list_inputs_or_outputsrArr@Úcontrolrr r¡r$)r+rvrwrxryrrr"r"r#rjs zDebugAnalyzer.list_inputsc CsB|jd |¡}t |¡}t |j¡}t |j¡\}}t   |¡\}} |j   ¡r||j   |¡s|t d|¡} t| dddd| S|j  |¡} | dkrtƒ} x$| D]} |  t|  d¡dƒ¡qžWt| ƒdkrØt| ƒd } n6d |t| ƒfd |g}t |¡} t| |ddd| Sg}x@| D]8} |j  | ¡}x$|D]}|j| kr.| |¡q.WqW|snt d |j¡} nÐt|ƒdkrà|jd kr¸tj|d  ¡|d j||j |||j!|j"d } nt d|j|jf¡} t| |ddn^|jd krÊd|jt|ƒfg}i}x†t#|ƒD]z\}}|j$|j j%d}| d|||jf¡d|j|f}t|dƒt|jƒt|dƒt &d|¡fg|t|ƒd<qW| d¡| d¡| d¡| d|j¡tj||d} nf|jt|ƒkrøt d|jt|ƒ|jf¡} n8tj||j ¡||jjd|j||j |||j"d} t| |dd| S)aTCommand handler for print_tensor. Print value of a given dumped tensor. Args: args: Command-line arguments, excluding the command prefix, as a list of str. screen_info: Optional dict input containing screen information such as cols. Returns: Output text lines as a RichTextLines object. rz,Node "%s" does not exist in partition graphsNTF)rrrú:rErz5Node "%s" generated debug dumps from %s output slots:z%Please specify the output slot: %s:x.z'Tensor "%s" did not generate any dumps.)rIÚtensor_slicingÚhighlight_optionsÚinclude_numeric_summaryÚ write_pathz³sz3DebugAnalyzer._make_source_table..rEcss|]}tt|dƒƒVqdS)éN)rtrP)r£rÑr"r"r#rÒµscss|]}tt|dƒƒVqdS)éN)rtrP)r£rÑr"r"r#rÒ¸sraNz ps %s -b %d) rZ COLOR_GRAYr¦Z COLOR_WHITErr rªrƒrtr Z%is_extension_uncompiled_python_sourcerrP)r+Ú source_listZis_tf_py_libraryZ path_headZnum_nodes_headZnum_tensors_headZnum_dumps_headÚcolorržZpath_column_widthZnum_nodes_column_widthZnum_tensors_column_widthÚheadr«rxZ num_nodesZ num_tensorsZ num_dumpsZfirst_line_numZpath_attributesrr"r"r#Ú_make_source_table’s^        z DebugAnalyzer._make_source_tablecCsØ~|jd |¡}tj|j|j|jd}tddƒg}|jrN| td|jƒ¡|jrh| td|jƒ¡| tƒ¡t   |¡}|s| d¡|S|  |  dd „|Dƒd ¡¡|  |  d d „|Dƒd ¡¡t |d d|S)z.FcSsg|]}|dr|‘qS)rEr")r£rÑr"r"r#r¤õsTN)r)rKrcr Zlist_source_files_against_dumpr'Z path_filterr7r¦rr rªrlrØr$)r+rvrwryrÕZ top_linesrr"r"r#rGÚs0      zDebugAnalyzer.list_sourceFc Cs6|r|jj}d}d} n|jj}d}d} g} i} t |¡\}} |j |¡sVt d|¡S|r`|} nd} |rrd| }nd}d ||f}t|ƒdt|ƒt|ƒdd fg| d <|   |d | |f¡|rÈd nd}|j | | ||| dg|||d |   d¡|   d¡|   d¡|r|   d¡|r(|   d¡t j | | dS)aùHelper function used by list_inputs and list_outputs. Format a list of lines to display the inputs or output recipients of a given node. Args: recursive: Whether the listing is to be done recursively, as a boolean. node_name: The name of the node in question, as a str. depth: Maximum recursion depth, applies only if recursive == True, as an int. control: Whether control inputs or control recipients are included, as a boolean. op_type: Whether the op types of the nodes are to be included, as a boolean. do_outputs: Whether recipients, instead of input nodes are to be listed, as a boolean. Returns: Input or recipient tree formatted as a RichTextLines object. z Recipients ofZ recipientsz Inputs torŸz3There is no node named "%s" in the partition graphsrEz, control %s includedr2z %s node "%s"rrz (Depth limit = %d%s):z lo -c -r %sz li -c -r %s)Úcommand_templatezLegend:z (d): recursion depth = d.z (Ctrl): Control input.z" [Op]: Input node has op type Op.)rb) r'ršr˜r r•r–rrgrtrÚ_dfs_from_noder rf)r+rArr@r°rr®ÚtrackerZtype_strZshort_type_strržrbrxÚ max_depthZinclude_ctrls_strrrÙr"r"r#r¯ùsT    (      z%DebugAnalyzer._list_inputs_or_outputsc  Cs| t ||dd¡¡} dgt| ƒ} |r^| t||ddƒ¡} |  | ¡|  dgt| ƒ¡| sx|dkrt| d¡dS| |¡d}xDt|ƒD]8}||dkrÀ|d|kr¶|t7}qÈ|t7}q|t 7}qW| rò||krò| |t ¡|  ¡dS|t |7}xüt | ƒD]ð\}}|j t |¡¡}||jkr2q| |rBt}nd}d}| rXt|}|t| ƒdkrr|  ¡||||}| |¡| rÆt|ƒt|ƒt|ƒt d| |¡fg|t|ƒd<t |¡\}}|j||||||d||| | d qWdS) a™Perform depth-first search (DFS) traversal of a node's input tree. It recursively tracks the inputs (or output recipients) of the node called node_name, and append these inputs (or output recipients) to a list of text lines (lines) with proper indentation that reflects the recursion depth, together with some formatting attributes (to attr_segs). The formatting attributes can include command shortcuts, for example. Args: lines: Text lines to append to, as a list of str. attr_segs: (dict) Attribute segments dictionary to append to. node_name: Name of the node, as a str. This arg is updated during the recursion. tracker: A callable that takes one str as the node name input and returns a list of str as the inputs/outputs. This makes it this function general enough to be used with both node-input and node-output tracking. max_depth: Maximum recursion depth, as an int. depth: Current recursion depth. This arg is updated during the recursion. unfinished: A stack of unfinished recursion depths, as a list of int. include_control: Whether control dependencies are to be included as inputs (and marked as such). show_op_type: Whether op type of the input nodes are to be displayed alongside the nodes' names. command_template: (str) Template for command shortcut of the node names. F)r“TrEz [None]Nr2)Úinclude_controlÚ show_op_typerÙ)r—ÚcopyrtrŒrlrrÚHANG_UNFINISHEDÚ HANG_FINISHEDÚ HANG_SUFFIXÚELLIPSISÚpopÚDEPTH_TEMPLATEr§r'ror r¡r¢Ú CTRL_LABELÚOP_TYPE_TEMPLATEr rr•rÚ)r+ržrrrÛrÜr@Ú unfinishedrÝrÞrÙZ all_inputsZis_ctrlr ZhangrÎrÅÚinprZctrl_strZ op_type_strrZ inp_node_namerxr"r"r#rÚTsl(              zDebugAnalyzer._dfs_from_nodec Cs:g}i}| d¡| dt|ƒ|t|ƒ|f¡| dt|ƒ|f¡x^|D]V}d|j |¡|f}| |¡t|ƒt|ƒt|ƒt dd|¡fg|t|ƒd<qLW|r,| d¡| dt|ƒ|f¡x^|D]V}d|j |¡|f}| |¡t|ƒt|ƒt|ƒt dd|¡fg|t|ƒd<qÒWtj||d S) aList neighbors (inputs or recipients) of a node. Args: neighbor_type: ("input" | "recipient") non_ctrls: Non-control neighbor node names, as a list of str. ctrls: Control neighbor node names, as a list of str. Returns: A RichTextLines object. r2z %d %s(s) + %d control %s(s):z %d %s(s):z [%s] %sNzni -a -d -t %srEz %d control %s(s):)rb)rrtr'ror rrf) r+Z neighbor_typeZ non_ctrlsZctrlsržrbZnon_ctrlrZctrlr"r"r#r™És,    $   $zDebugAnalyzer._format_neighborscCszg}| d¡| d¡|j |¡}xJ|D]B}| d|¡t||ƒ ¡ dd¡}| d|¡| d¡q*Wt |¡S)z¸List neighbors (inputs or recipients) of a node. Args: node_name: Name of the node of which the attributes are to be listed. Returns: A RichTextLines object. r2zNode attributes:z %s:Ú raz %s)rr'Znode_attributesÚreprÚstripÚreplacer rf)r+rržÚattrsZattr_keyZ attr_val_strr"r"r#r›ós     z#DebugAnalyzer._list_node_attributesc CsÔg}i}|j |¡}d}xŒ|D]„}|j |¡}xr|D]j}d|j|j|j|jjdf} | | ¡d||j|f} dt| ƒt   d| ¡fg|t|ƒd<|d7}q4WqWt j ||d} t   d |d g¡} |   | ¡| S) z¬List dumped tensor data from a node. Args: node_name: Name of the node of which the attributes are to be listed. Returns: A RichTextLines object. rz Slot %d @ %s @ %.3f msg@@zpt %s:%d -n %drÓNrE)rbz%d dumped tensor(s):r2) r'r¸r½rsZdebug_oprrprrtr rrfrl) r+rržrbrÂr}rÀrÃrÄrr’rZoutput_with_headerr"r"r#rœ s(      $  zDebugAnalyzer._list_node_dumps)N)N)N)N)N)N)r)N)N)F)FFN)"Ú__name__Ú __module__Ú __qualname__Ú__doc__r„r…r†rŽr¢r.r*r]r^r`rrjrkrmrr—rrrrrÇrËrFrØrGr¯rÚr™r›rœr"r"r"r#r%‚sH( w)#@ U# $  $   ;H % ] k*r%Úcursesc Csn|dkrt ¡}t||d}|r>x|D]}| |||¡q&Wtj|||d}|jd|j| d¡dgd|jd|j | d¡dgd|jd |j | d ¡d gd|jd |j | d ¡d gd|jd |j | d ¡dgd|jd|j | d¡dgd|jd|j| d¡dgd|jd|j| d¡dgdg}x&|jD]} | d| j| jf¡q:W| d dg|¡|S)a2Create an instance of CursesUI based on a DebugDumpDir object. Args: debug_dump: (debug_data.DebugDumpDir) The debug dump to use. tensor_filters: (dict) A dict mapping tensor filter name (str) to tensor filter (Callable). ui_type: (str) requested UI type, e.g., "curses", "readline". on_ui_exit: (`Callable`) the callback to be called when the UI exits. config: A `cli_config.CLIConfig` object. Returns: (base_ui.BaseUI) A BaseUI subtype object with a set of standard analyzer commands and tab-completions registered. N)r-)Ú on_ui_exitr-rÚlt)Zprefix_aliasesrÚnirÚlirÚlorÚptrFZpsrGZlsrJZevz%s:%d)rZ CLIConfigr%r]r Zget_uiZregister_command_handlerrr`rrrrrFrGrÇrirrrsZregister_tab_comp_context) r,Ztensor_filtersZui_typerôr-ZanalyzerZtensor_filter_nameÚcliZdumped_tensor_namesrÄr"r"r#Úcreate_analyzer_ui.sj           rû)NTTTTT)NróNN)%ròÚ __future__rrrrLrßrdZ six.movesrZtensorflow.python.debug.clirrrr r r Ztensorflow.python.debug.libr r ZRichLiner¦ràrárârårçrærãrQrRrSrTr$Úobjectr%rûr"r"r"r#Úsb             ?8