B һd.@s.dZddlZd ddZd dd Zdd d ZdS)z&TensorFlow Debugger (tfdbg) Utilities.N DebugIdentityFc Csj|jj}||j_|}||_||_||_t|tr:|g}|j ||rft|trZ|g}|j |dS)aAdd watch on a `Tensor` to `RunOptions`. N.B.: 1. Under certain circumstances, the `Tensor` may not get actually watched (e.g., if the node of the `Tensor` is constant-folded during runtime). 2. For debugging purposes, the `parallel_iteration` attribute of all `tf.while_loop`s in the graph are set to 1 to prevent any node from being executed multiple times concurrently. This change does not affect subsequent non-debugged runs of the same `tf.while_loop`s. Args: run_options: An instance of `config_pb2.RunOptions` to be modified. node_name: (`str`) name of the node to watch. output_slot: (`int`) output slot index of the tensor from the watched node. debug_ops: (`str` or `list` of `str`) name(s) of the debug op(s). Can be a `list` of `str` or a single `str`. The latter case is equivalent to a `list` of `str` with only one element. For debug op types with customizable attributes, each debug op string can optionally contain a list of attribute names, in the syntax of: debug_op_name(attr_name_1=attr_value_1;attr_name_2=attr_value_2;...) debug_urls: (`str` or `list` of `str`) URL(s) to send debug values to, e.g., `file:///tmp/tfdbg_dump_1`, `grpc://localhost:12345`. tolerate_debug_op_creation_failures: (`bool`) Whether to tolerate debug op creation failures by not throwing exceptions. global_step: (`int`) Optional global_step count for this debug tensor watch. N) debug_optionsZdebug_tensor_watch_opts global_stepadd#tolerate_debug_op_creation_failures node_name output_slot isinstancestr debug_opsextend debug_urls) run_optionsrr r rrrZ watch_optswatchrY/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/debug/lib/debug_utils.pyadd_debug_tensor_watchs#   rc  Cs.|s td|stdt|tr(|g}|r6t|nd} |rHt|nd} |rZt|nd} |} x| D]}|jsxql|j}|j}| r| |sql| r| |sqlxFt t |jD]4}| r| |j|j jsqt |||||||dqWqlW| o| o| }|r"t |dd||||d| |j_dS)aa Add debug watches to `RunOptions` for a TensorFlow graph. To watch all `Tensor`s on the graph, let both `node_name_regex_allowlist` and `op_type_regex_allowlist` be the default (`None`). N.B.: 1. Under certain circumstances, the `Tensor` may not get actually watched (e.g., if the node of the `Tensor` is constant-folded during runtime). 2. For debugging purposes, the `parallel_iteration` attribute of all `tf.while_loop`s in the graph are set to 1 to prevent any node from being executed multiple times concurrently. This change does not affect subsequent non-debugged runs of the same `tf.while_loop`s. Args: run_options: An instance of `config_pb2.RunOptions` to be modified. graph: An instance of `ops.Graph`. debug_ops: (`str` or `list` of `str`) name(s) of the debug op(s) to use. debug_urls: URLs to send debug values to. Can be a list of strings, a single string, or None. The case of a single string is equivalent to a list consisting of a single string, e.g., `file:///tmp/tfdbg_dump_1`, `grpc://localhost:12345`. For debug op types with customizable attributes, each debug op name string can optionally contain a list of attribute names, in the syntax of: debug_op_name(attr_name_1=attr_value_1;attr_name_2=attr_value_2;...) node_name_regex_allowlist: Regular-expression allowlist for node_name, e.g., `"(weight_[0-9]+|bias_.*)"` op_type_regex_allowlist: Regular-expression allowlist for the op type of nodes, e.g., `"(Variable|Add)"`. If both `node_name_regex_allowlist` and `op_type_regex_allowlist` are set, the two filtering operations will occur in a logical `AND` relation. In other words, a node will be included if and only if it hits both allowlists. tensor_dtype_regex_allowlist: Regular-expression allowlist for Tensor data type, e.g., `"^int.*"`. This allowlist operates in logical `AND` relations to the two allowlists above. tolerate_debug_op_creation_failures: (`bool`) whether debug op creation failures (e.g., due to dtype incompatibility) are to be tolerated by not throwing exceptions. global_step: (`int`) Optional global_step count for this debug tensor watch. reset_disk_byte_usage: (`bool`) whether to reset the tracked disk byte usage to zero (default: `False`). z$debug_ops must not be empty or None.z%debug_urls must not be empty or None.N)r r rrr*r) ValueErrorr r recompileget_operationsoutputsnametypematchrangelendtyperrreset_disk_byte_usage)rgraphr rZnode_name_regex_allowlistZop_type_regex_allowlistZtensor_dtype_regex_allowlistrrr node_name_patternop_type_patterntensor_dtype_patternopsoprop_typeslotZuse_node_name_wildcardrrr watch_graphMsX7  r)c  Cst|tr|g}|rt|nd} |r0t|nd} |rBt|nd} |} x| D]}|js`qT|j}|j}| r|| |r|qT| r| |rqTxFt t |jD]4}| r| |j|j jrqt |||||||dqW| |j _qTWdS)aYAdd debug tensor watches, denylisting nodes and op types. This is similar to `watch_graph()`, but the node names and op types are denylisted, instead of allowlisted. N.B.: 1. Under certain circumstances, the `Tensor` may not get actually watched (e.g., if the node of the `Tensor` is constant-folded during runtime). 2. For debugging purposes, the `parallel_iteration` attribute of all `tf.while_loop`s in the graph are set to 1 to prevent any node from being executed multiple times concurrently. This change does not affect subsequent non-debugged runs of the same `tf.while_loop`s. Args: run_options: An instance of `config_pb2.RunOptions` to be modified. graph: An instance of `ops.Graph`. debug_ops: (`str` or `list` of `str`) name(s) of the debug op(s) to use. See the documentation of `watch_graph` for more details. debug_urls: URL(s) to send debug values to, e.g., `file:///tmp/tfdbg_dump_1`, `grpc://localhost:12345`. node_name_regex_denylist: Regular-expression denylist for node_name. This should be a string, e.g., `"(weight_[0-9]+|bias_.*)"`. op_type_regex_denylist: Regular-expression denylist for the op type of nodes, e.g., `"(Variable|Add)"`. If both node_name_regex_denylist and op_type_regex_denylist are set, the two filtering operations will occur in a logical `OR` relation. In other words, a node will be excluded if it hits either of the two denylists; a node will be included if and only if it hits neither of the denylists. tensor_dtype_regex_denylist: Regular-expression denylist for Tensor data type, e.g., `"^int.*"`. This denylist operates in logical `OR` relations to the two allowlists above. tolerate_debug_op_creation_failures: (`bool`) whether debug op creation failures (e.g., due to dtype incompatibility) are to be tolerated by not throwing exceptions. global_step: (`int`) Optional global_step count for this debug tensor watch. reset_disk_byte_usage: (`bool`) whether to reset the tracked disk byte usage to zero (default: `False`). N)r r rrr)r r rrrrrrrrrrrrr )rr!r rZnode_name_regex_denylistZop_type_regex_denylistZtensor_dtype_regex_denylistrrr r"r#r$r%r&rr'r(rrrwatch_graph_with_denylistss81   r*)rrNFr)rNNNNFrF)rNNNNFrF)__doc__rrr)r*rrrrs, 4 q