BVh6< TdZddlZddlZddlZddlZddlZddlmZdZdededZ de d Z eje ejZ ejd d d gZej j#ej j#ej j#eZej j)ed ej*zej j)edej*zej j)ej j#eddej*zgZejdgZ ej1ej2ej4gzDcgc]+}ej j)|dej*z-c}ejdgZdZddZddZ ddZ!ddZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)ycc}w#e$r$ej9ejdYiwxYw) zFunction for interpolating formatted errors from the TensorFlow runtime. Exposes the function `interpolate` to interpolate messages with tags of the form {{type name}}. N)tf_stackz[A-Za-z0-9_.][A-Za-z0-9_.\-/]*?z {{(?Pz ) (?Pz)}}z(?P.*?)(?P) _ParseTagtypenamepythoncontribpykeras with tf.device(some_func): The first line will have no padding to its left by default. Subsequent lines will have two spaces of left-padding. Use the prefix argument to increase indentation. z:No device assignments were active during op '%s' creation.z%Device assignments active during op ' ' creation:<{file}:{line}>fileline )prefixindentdev_namelocz1{prefix}{indent}with tf.device({dev_name}): {loc} )rformatfilenamelinenoobjr)rdevice_assignment_listr)rstr_list traceable_objlocation_summarysubss r!_compute_device_summary_from_listr8ts$ JG tOG G  ( //@FMO. Lm(//  # #-*>*>0@!%%  D  OOB;BBJTJL L 8 r!cDt|j|j|S)N)r8r_device_assignmentsopr)s r*_compute_device_assignment_summary_from_opr=s *277B4J4J+1 33r!c@|s d}||z}||zSg}|j|d|d|jD]T\}}dj|j|j}|d||d}|jdjd i|Vd j |S) aReturn a summary of an op's colocation stack. Args: name: The op name. colocation_dict: The op._colocation_dict. prefix: An optional string prefix used before each line of the multi- line string returned by this function. Returns: A multi-line string similar to: Node-device colocations active during op creation: with tf.compat.v1.colocate_with(test_node_1): with tf.compat.v1.colocate_with(test_node_2): The first line will have no padding to its left by default. Subsequent lines will have two spaces of left-padding. Use the prefix argument to increase indentation. z?No node-device colocations were active during op '%s' creation.z*Node-device colocations active during op 'r#r$r%r()r)r*rr,z4{prefix}{indent}with tf.colocate_with({name}): {loc}r-r.)ritemsr/r0r1r) rcolocation_dictr)rr4 coloc_namelocationr6r7s r%_compute_colocation_summary_from_dictrCs$ OG tOG G  ( //4!".335 Oj((//   X__06  D  OOE>EEMMO O 8 r!cDt|j|j|S)zEFetch colocation file, line, and nesting and return a summary string.)rCr_colocation_dictr;s r#_compute_colocation_summary_from_oprFs" /rww8K8K/5 77r!ctD]}|j|sytD]}|j|sytD]}|j |syy)aReturns whether a filename should be considered a part of the framework. A file is part of the framework if it does not match a pattern in _EXTERNAL_FILENAME_PATTERNS and it either matches a pattern in _FRAMEWORK_FILENAME_PATTERNS or starts with a _FRAMEWORK_PATH_PREFIXES prefix. Args: filename: A filename string. Returns: Whether the filename should be considered to be internal to the TensorFlow framework for the purposes of reporting errors. FT)_EXTERNAL_FILENAME_PATTERNSsearch_FRAMEWORK_FILENAME_PATTERNS_FRAMEWORK_PATH_PREFIXES startswith)r0patternr)s r_is_framework_filenamerNsi-g~~h .g~~h )f6"  r!ct|}|Dcgc]}|j}}tt|D]\}}t |}|r||z dz cSycc}w)aNReturn index in op.traceback with first 'useful' frame. This method reads through the stack stored in op.traceback looking for the innermost frame which (hopefully) belongs to the caller. It accomplishes this by rejecting frames deemed to be part of the TensorFlow framework (by pattern matching the filename). Args: tb: A list of traceback frames (as from Operation.traceback). Returns: Integer index into op.traceback where the first non-TF file was found (innermost to outermost), or 0 (for the outermost stack frame) if all files came from TensorFlow. r)lenr0 enumeratereversedrN)tbsizeframe filenamesidxr0 is_frameworks r_find_index_of_defining_framerZsf" R$+-.%u~~.). )!45mc8)(3L  CZ!^  /sAcvt|}t|dzdzt|}t||z d}|||S)awReturn a list of frames, which form a 'useful' stack. Starting from the defining frame to the outermost one, this method computes the contiguous portion of the 'useful' stack trace and returns the selected frames. Args: tb: A list of traceback frames (as from Operation.traceback). num: total number of frames to return. Returns: A list of frames. rPr)rZminrQmax)rTnumdefining_frame_indexinnermost_excludedoutermost_includeds r_compute_useful_framesrcsO7r: /!3a7RA-3Q7 1 22r!c tj}|D]C\}}|j|j||jt |jdE|j S)aConstruct and returns a `GraphDebugInfo` protocol buffer. Args: func_named_operations: An iterable of (func_name, op.Operation) tuples where the Operation instances have a _traceback members. The func_name should be the empty string for operations in the top-level Graph. Returns: GraphDebugInfo protocol buffer. Raises: TypeError: If the arguments are not of the correct proto buffer type. )rGraphDebugInfoBuilder tracebackAccumulateStackTracerrcBuild)func_named_operationsbuilder func_namer<s rcreate_graph_debug_info_defrm'se  * * ,',mi ||   27722<<D r!c t|}t|}dj||g}|j d}d}d}d}d}ns|jj } | j }tj |jj}| j}| j}|d|d}||||||||d} | S)a%Return a dictionary mapping interpolation tokens to values. Args: op: op.Operation object. Returns: A dictionary mapping string tokens to string values. The keys are shown below along with example values. { "file": "tool_utils.py", "lineno": "124", "line": " source code line", "defined_at": " (defined at tool_utils.py:124)", "colocations": '''Node-device colocations active during op creation: with tf.compat.v1.colocate_with(test_node_1): with tf.compat.v1.colocate_with(test_node_2): ''' "devices": '''Device assignments active during op 'foo' creation: with tf.device(/cpu:0): with tf.device(some_func): ''' "devs_and_colocs": A concatenation of colocations and devices, e.g. '''Node-device colocations active during op creation: with tf.compat.v1.colocate_with(test_node_1): with tf.compat.v1.colocate_with(test_node_2): ''' Device assignments active during op 'foo' creation: with tf.device(/cpu:0): with tf.device(some_func): ''' } r-z rr:d) colocationsdevicesdevs_and_colocs defined_atr&r1r'definition_traceback) rFr=rrglast_user_framer0 format_listget_user_framesr1r') r<colocation_summarydevice_summarycombined_summaryr0rur1r'rtrV field_dicts r_compute_field_dictr}@sB;2>=bA.YY 2NCD\\H F DJ LL ( ( *E~~H$001M1M1OP \\F ::D:Qvaj)J()2 * r!cDd|jdg}t|}|dD]I}d|vs|j|jdDcgc]}|j sd|c}K|j d|jdj |Scc}w)zReturns the formatted error message for the given op. Args: op: The node. Returns: The formatted error message for the given op with traceback. zDetected at node z$ defined at (most recent call last):rur r-r(zNode: )rr}extendsplitstriprr)r<node_error_messager|rVr's r_build_node_error_messagers "''$HI#2&*01Ge%#(;;t#4 E4 Rv; EGG fRWWK01 % && Fs B B c t|\}}}ddg}|D]8} |j|j}|jt |:|j|j dj|S#t $rYvwxYw)aInterpolates an error message. The error message can contain tags of form `{{node_type node_name}}` which will be parsed to identify the tf.Graph and op. If the op contains traceback, the traceback will be attached to the error message. Args: message: A string to interpolate. graph: ops.Graph object containing all nodes referenced in the error message. Returns: The error message string with node definition traceback. zGraph execution error:rr-)r get_operation_by_namerrrKeyErrorrr)rgraphparsed_messaged_rrtagr<s rinterpolate_graphrs#0"8/1i+R0- :c:  & &sxx 0b4R89 :,,./ = !! sB B  B )r)*__doc__ collectionsosrsitergtensorflow.python.utilr _NAME_REGEX _TAG_REGEX_INTERPOLATION_REGEXcompileDOTALLr namedtuplerpathdirname__file___FRAMEWORK_COMMON_PREFIXrrrKrJrgetsitepackagesgetusersitepackagesAttributeErrorrrHr r8r=rCrFrNrZrcrmr}rr) package_paths0rrs  +0 k]+k]% H .zl!<#$8"))D "K " ";0@ A 77??GGOOBGGOOH-.0GGLL)84rvv=GGLL)95>GGLL!9:w "$&&)BJJ| <!!.$..04LD4L4L4N3OO#  ggll<)BFF2#BJJ} 6F'T3'T74 <302?D'4"y #<%%jbjj&:;0G96G>9G>>&H'&H'