B )`_@sdZddlmZddlmZddlmZddlZddlZddlm Z e dZ e dZ e d Z d Zd d ZGd ddeZdS)zJLibrary for arbitrary expression evaluation based on a debugger data dump.)absolute_import)division)print_functionN) debug_dataz`.*?`z0/job:(\w)+/replica:(\d)+/task:(\d)+/(\w)+:(\d)+:z \[(\d)*\]$Z DebugIdentityc Cstt|}|r:|||d}||d}nd}|d}t|dkr`td|tt |d}|rt |d|d|d}|dd||d<nd}t|dkr|d}t |d}t }n&|d}|d}t |d}|d}|||||fS) a}Parse a debug tensor name in a to-be-evaluated expression. Args: debug_tensor_name: name of the debug tensor, with or without device name as a prefix, with or without debug op, with or without '[]' as a suffix. E.g., without device name prefix, without debug op suffix: "hidden_0/MatMul:0" E.g., with device name prefix: "/job:worker/replica:0/task:1/gpu:0:hidden_0/MatMul:0" E.g., with debug op suffix: "hidden_0/MatMul:0:DebugNumericSummary" E.g., with device name prefix and debug op suffix: "/job:worker/replica:0/task:1/gpu:0:hidden_0/MatMul:0:DebugNumericSummary" E.g., with device name prefix, debug op and an exec index: "/job:worker/replica:0/task:1/gpu:0:hidden_0/MatMul:0:DebugNumericSummary[1]" Returns: device_name: If device name prefix exists, the device name; otherwise, `None`. node_name: Name of the node. output_slot: Output slot index as an `int`. debug_op: If the debug op suffix exists, the debug op name; otherwise, `None`. exec_index: Execution index (applicable to cases in which a debug tensor is computed multiple times in a `tf.Session.run` call, e.g., due to `tf.while_loop`). If the exec_index suffix does not exist, this value defaults to `0`. Raises: ValueError: If the input `debug_tensor_name` is malformed. N:)zJThe debug tensor name in the to-be-evaluated expression is malformed: '%s'rr) rematch_DEVICE_NAME_PREFIX_PATTERNstartendsplitlen ValueErrorsearch_EXEC_INDEX_SUFFIX_PATTERNint_DEFAULT_DEBUG_OP) Zdebug_tensor_nameZdevice_prefix_match device_nameZ split_itemsZexec_index_match exec_index node_name output_slotdebug_oprT/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/python/debug/cli/evaluator.py_parse_debug_tensor_name"s4#       rc@s eZdZdZddZddZdS)ExpressionEvaluatorzCEvaluates Python expressions using debug tensor values from a dump.cCs||_i|_dS)z^Constructor of ExpressionEvaluator. Args: dump: an instance of `DebugDumpDir`. N)_dump_cached_tensor_values)selfdumprrr__init__mszExpressionEvaluator.__init__c Cstt|}|}xtt|D]}|ddd}t|\}}}} } ||jkry|j j ||| |d| } Wn&t j k rt d||fYnX| |j|<|d|dd|d||dd}qWt|S) aParse an expression. Args: expression: the expression to be parsed. Returns: The result of the evaluation. Raises: ValueError: If the value of one or more of the debug tensors in the expression are not available. rrr )rzEEval failed due to the value of %s:%d:DebugIdentity being unavailableNzself._cached_tensor_values['z'])r finditer_DUMP_TENSOR_PATTERNreversedlistgroupstriprr!r Z get_tensorsrZ'WatchKeyDoesNotExistInDebugDumpDirErrorrrreval) r"Z expressionZdump_tensors_iterZrewritten_expressionr Z tensor_namerrrrrvaluerrrevaluatevs$   zExpressionEvaluator.evaluateN)__name__ __module__ __qualname____doc__r$r-rrrrrjs r)r1 __future__rrrr numpynpZtensorflow.python.debug.librcompiler&r rrrobjectrrrrrs      H