BVh >.dZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZej$Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZ eee gZ!d$dZ"d%dZ#dZ$dZ% d&dZ&dZ'd'dZ(dZ)d Z* d$d!Z+ d$d"Z,d#Z-y)(z>Shared functions and classes for tfdbg command-line interface.N)command_parser)debugger_cli_common) tensor_format)common)ops)tensor) variables)gfileiblackbluecyangraygreenmagentaredwhiteyellowusmssc| t|S|dkrd|z}n$|dkr d|dz z}n|dkr d|dz z}nd |d z z}|r|d z }|S) aeGenerate a human-readable string representing number of bytes. The units B, kB, MB and GB are used. Args: num_bytes: (`int` or None) Number of bytes. include_b: (`bool`) Include the letter B at the end of the unit. Returns: (`str`) A string representing the number of bytes in a human-readable way, including a unit at the end. iz%diz%.2fkg@i@z%.2fMg0Az%.2fGgAB)str) num_bytes include_bresults V/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/debug/cli/cli_shared.pybytes_to_readable_strr3st y> I F7  F* +F:  I- .F  L0 1F cMF -c |sy|rY|tvrtd|ztj|}|}dj|t j dd|zz |St ttdz tt j|ddz }t|}dj|t j dd|zz |S) aWConvert time value to human-readable string. Args: value_us: time value in microseconds. force_time_unit: force the output to use the specified time unit. Must be in TIME_UNITS. Returns: Human-readable string representation of the time value. Raises: ValueError: if force_time_unit value is not in TIME_UNITS. 0zInvalid time unit: %sz {:.10g}{}g$@ z{:.3g}{}) TIME_UNITS ValueErrorindexformatmathpowminlenintlog)value_usforce_time_unitorder time_units rtime_to_readable_strr3Qs  j( .@ AA   _ -EI   h$%)@@) LL J!#S(B)?!)C%D EE5!I   Xqw(?? KKrcpdfd}|r,tj|tj||Sy)avProcess ranges highlight string. Args: ranges_string: (str) A string representing a numerical range of a list of numerical ranges. See the help info of the -r flag of the print_tensor command for more details. Returns: An instance of tensor_format.HighlightOptions, if range_string is a valid representation of a range or a list of ranges. Nc tj|jt}D]5\}}tj|tj ||k\||k}7|S)N)dtype)npzerosshapebool logical_or logical_and)xr range_start range_endrangess r ranges_filterz-parse_ranges_highlight..ranges_filter|sX %A"(M Y --2>>!{*:ANK LaM Hr) description)r parse_rangesrHighlightOptions) ranges_stringrBrAs @rparse_ranges_highlightrGmsA &   ( ( 7F  ) )= 22 rc |r d|vrd|diSiS)Ncols linewidth) screen_infos r#numpy_printoptions_from_screen_inforMs!V{* V, -- Irc L|rtj||}||z} n|}|} d} |rtj|d5} t j | |dddt jd} | t j|dz } | dttj|jzz } t j| t jdg} |r|j|d<n t|d<tj|| d || || S#1swYxYw) aGenerate formatted str to represent a tensor or its slices. Args: tensor: (numpy ndarray) The tensor value. tensor_name: (str) Name of the tensor, e.g., the tensor's debug watch key. np_printoptions: (dict) Numpy tensor formatting options. print_all: (bool) Whether the tensor is to be displayed in its entirety, instead of printing ellipses, even if its number of elements exceeds the default numpy display threshold. (Note: Even if this is set to true, the screen output can still be cut off by the UI frontend if it consist of more lines than the frontend can handle.) tensor_slicing: (str or None) Slicing of the tensor, e.g., "[:, 1]". If None, no slicing will be performed on the tensor. highlight_options: (tensor_format.HighlightOptions) options to highlight elements of the tensor. See the doc of tensor_format.format_tensor() for more details. include_numeric_summary: Whether a text summary of the numeric values (if applicable) will be included. write_path: A path to save the tensor value (after any slicing) to (optional). `numpy.save()` is used to save the value. Returns: An instance of `debugger_cli_common.RichTextLines` representing the (potentially sliced) tensor. NwbzSaved value to: bold) font_attrz (%sB) thresholdT)include_metadatainclude_numeric_summaryauxiliary_messagenp_printoptionshighlight_options)revaluate_tensor_slicer Openr7saverRichLinerStatlength#rich_text_lines_from_rich_line_listsize!DEFAULT_NDARRAY_DISPLAY_THRESHOLDr format_tensor) r tensor_namerW print_alltensor_slicingrXrU write_pathvalue sliced_namerV output_filelines rrbrbs!F  0 0 HE.K EK J %"ggk5!"  ' '(: ;D  ( (v FFDH,UZZ -C-J-JK KKD+OO "++B/02#(::OK #DOK  $ $ 5)%) ++""s DD#cPtjtd|ztgS)zGenerate a RichTextLines output for error. Args: msg: (str) The error message. Returns: (debugger_cli_common.RichTextLines) A representation of the error message for screen output. zERROR: )rr_RL COLOR_RED)msgs rerrorros+  @ @S)$B& ''rcd|z}|rtjd|dg}nd}t|t||zdz|dz|zg}tj|S)aGenerate a RichTextLines object that describes a recommended command. Args: command: (str) The command to recommend. description: (str) A description of what the command does. indent: (int) How many spaces to indent in the beginning. create_link: (bool) Whether a command link is to be applied to the command string. Returns: (RichTextLines) Formatted text (with font attributes) for recommending the command.  rRrP: )rMenuItemrlr_)commandrCindent create_link indent_strrQliness r_recommend_commandrzskV|*$--b':FCII j>Bw 2 2S 8  { * ,%  @ @ GGrc4gd}tj|S)z/Make an ASCII representation of the tfdbg logo.)rRzTTTTTT FFFF DDD BBBB GGG z TT F D D B B G z TT FFF D D BBBB G GGz TT F D D B B G Gz TT F DDD BBBB GGG rR)r RichTextLines)rys rget_tfdbg_logor}s %  * *5 11rz&======================================c Btj|}|stjdg}ntg}|D]m}tj|}tjd} | tj|tj dd|zz } |j | otj|}tjt} |r| j dn| j d|z| j d| j d| jtj|D cgc]} d| z c} | j d| j d | j|| j t| j d| j d | jtd d d | jtddd | jtdd| jtdddg} |r[g} |D]S}| j |tj dd|z}| j tdt||zUn| j d| jtj| | j d| jtdtdtj ddzdz| j dtj}|j tj d d |j tj dd|| jtj <| Scc} w)aGGenerate formatted intro for run-start UI. Args: run_call_count: (int) Run call counter. fetches: Fetches of the `Session.run()` call. See doc of `Session.run()` for more details. feed_dict: Feeds to the `Session.run()` call. See doc of `Session.run()` for more details. tensor_filters: (dict) A dict from tensor-filter name to tensor-filter callable. is_callable_runner: (bool) whether a runner returned by Session.make_callable is being run. Returns: (RichTextLines) Formatted intro message about the `Session.run()` call. z (Empty)rsNzpf '%s'z4Running a runner returned by Session.make_callable()zSession.run() call #%d:rRz Fetch(es):z Feed dict:z5Select one of the following commands to proceed ---->runz1Execute the run() call with debug tensor-watchingTrwzrun -nz4Execute the run() call without debug tensor-watchingz run -t zyExecute run() calls (T - 1) times without debugging, then execute run() once more with debugging and drop back to the CLIzrun -f zpKeep executing run() calls until a dumped tensor passes a given, registered filter (conditional breakpoint mode)z Registered filter(s):z run -f %sz * z (None)zFor more details, see zhelp.help.exit)rget_flattened_namesrr\get_graph_element_namertappendr_r|_HORIZONTAL_BARextendrzrlappend_rich_lineMenu annotations MAIN_MENU_KEY)run_call_countfetches feed_dicttensor_filtersis_callable_runner fetch_linesfeed_dict_linesfeed_key feed_key_namefeed_dict_lineoutrj more_lines filter_names filter_namecommand_menu_nodemenus rget_run_start_intrors@,**73+ *33K@AOO-33H=m*33D9n+44   & &tY-F GIIn ^,-(KK/ ))/:#JJEFJJ(>9:JJrNJJ|JJ"00!,--/0JJrNJJ|JJ**_**R.** DE**  = **  @ **  LMN **  <=> ,,*L%O +&-66 kK')<(2k;L+MMN O &'**==jIK**R.r23'#6#?#?F#KLM**R.  ! ! #$++!**5%89++!**66:;7;#//%334 *q .s9 N c|ryd|z}t|tjtjt j fr|dtj|zz }n4ttj|}|dkDr |d|zz }n|d|zz }|s|dz }|St|dk(r9|D]2}|dt|ts t|d s|n |jzz }4|S|d t|zz }|S) aGet a short description of the run() call. Args: run_call_count: (int) Run call counter. fetches: Fetches of the `Session.run()` call. See doc of `Session.run()` for more details. feed_dict: Feeds to the `Session.run()` call. See doc of `Session.run()` for more details. is_callable_runner: (bool) whether a runner returned by Session.make_callable is being run. Returns: (str) A short description of the run() call, including information about the fetche(s) and feed(s). zrunner from make_callable()z run #%d: z1 fetch (%s); r#z %d fetches; z %d fetch; z0 feedsz 1 feed (%s)namez%d feeds) isinstance tensor_libTensorr Operationr Variablerrr,rrhasattrr)rrrrrC num_fetcheskeys rget_run_short_descriptionr~s& (n,+ !!3==)2D2DE#f&C&CG&LLLKf009:KQ^k11k\K//k 9K  9~O#}#s#73+? EHXXO O O Z#i.00k rc t|dr-t|jdr|jj}nd}dtdddg}t j |}||j t jdg|j td |zd d |j td |zdd |j tddd n%|j t jdgdd|zdtt|zddt|dddg }|j t j||S)aGenerate formatted intro for TensorFlow run-time error. Args: tf_error: (errors.OpError) TensorFlow run-time error object. Returns: (RichTextLines) Formatted intro message about the run-time OpError, with sample commands for debugging. oprNz&--------------------------------------z(!!! An error occurred during the run !!!blinkrRz,You may use the following commands to debug:zni -a -d -t %sz)Inspect information about the failing op.Trzli -r %sz+List inputs to the failing op, recursively.ltz6List all tensors dumped during the failing run() call.zCWARNING: Cannot determine the name of the op that caused the error.zOp name: %sz Error type: zDetails:) rrrrlrr_rr|rzrtype)tf_errorop_name intro_linesrrs rget_error_introrsf Xtf!=kkGG/ 3W=+ ?? L# JJ"00 78:;JJ+g5F'+ -.JJ:/H'+ -. JJ  D  JJ"00M2OPQ  s4>** (m. *** . .z :; *r)F)N)FNNFN)F).__doc__r)numpyr7tensorflow.python.debug.clirrrtensorflow.python.debug.librtensorflow.python.frameworkrrrtensorflow.python.opsr tensorflow.python.platformr r\rlra COLOR_BLACK COLOR_BLUE COLOR_CYAN COLOR_GRAY COLOR_GREEN COLOR_MAGENTArm COLOR_WHITE COLOR_YELLOW TIME_UNIT_US TIME_UNIT_MS TIME_UNIT_Sr%rr3rGrMrbrorzr}rrrrrKrrrsE 6;5.+<+,!!%)!            L+ 6 <L8<"!%$(*/!A+H 'H8 2;,1 h \27/d; r