o i e.@sXdZddlZ     d ddZ        d dd Z        d d d ZdS)z&TensorFlow Debugger (tfdbg) Utilities.N DebugIdentityFc Csn|jj}||j_|}||_||_||_t|tr|g}|j ||r5t|tr-|g}|j |dSdS)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_optionsdebug_tensor_watch_opts global_stepadd#tolerate_debug_op_creation_failures node_name output_slot isinstancestr debug_opsextend debug_urls) run_optionsr r r rrr watch_optswatchrV/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/debug/lib/debug_utils.pyadd_debug_tensor_watchs#   rc  Cs |std|s tdt|tr|g}|rt|nd} |r$t|nd} |r-t|nd} |} | D]>}|js;q5|j}|j}| rI| |sIq5| rQ| |sQq5t t |jD]}| rg| |j|j jsgqXt |||||||dqXq5| o|| o|| }|rt |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.Nr r rrr*r) ValueErrorr r recompileget_operationsoutputsnametypematchrangelendtyperrreset_disk_byte_usage)rgraphr rnode_name_regex_allowlistop_type_regex_allowlisttensor_dtype_regex_allowlistrrr#node_name_patternop_type_patterntensor_dtype_patternopsopr op_typeslotuse_node_name_wildcardrrr watch_graphMsr7    r0c  Cst|tr|g}|rt|nd} |rt|nd} |r!t|nd} |} | D]B}|js/q)|j}|j}| r=| |r=q)| rE| |rEq)t t |jD]}| r[| |j|j jr[qLt |||||||dqL| |j _q)dS)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`). Nr)r r rrrrrrrr r!r"rrr#)rr$r rnode_name_regex_denylistop_type_regex_denylisttensor_dtype_regex_denylistrrr#r(r)r*r+r,r r-r.rrrwatch_graph_with_denylistssL 1   r4)rrNFr)rNNNNFrF)__doc__rrr0r4rrrrs4 : z