BVh4dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z Gd d e jZ Gd d e jZGd de jZGddeZy)ztfdbg CLI as SessionRunHook.) config_pb2) debug_utils)dumping_wrapper) framework) grpc_wrapper)local_cli_wrapper)session_run_hookc8eZdZdZ ddZdZdZdZdZy) LocalCLIDebugHookzlCommand-line-interface debugger hook. Can be used as a hook for `tf.compat.v1.train.MonitoredSession`. NcX||_||_||_d|_i|_||_y)aCreate a local debugger command-line interface (CLI) hook. Args: ui_type: (`str`) requested user-interface type. Currently supported: (readline). dump_root: (`str`) optional path to the dump root directory. Must be a directory that does not exist or an empty directory. If the directory does not exist, it will be created by the debugger core during debug `run()` calls and removed afterwards. thread_name_filter: Regular-expression white list for threads on which the wrapper session will be active. See doc of `BaseDebugWrapperSession` for more details. config_file_path: Optional override to the default configuration file path, which is at `${HOME}/.tfdbg_config`. N)_ui_type _dump_root_thread_name_filter_session_wrapper_pending_tensor_filters_config_file_path)selfui_type dump_rootthread_name_filterconfig_file_paths V/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/debug/wrappers/hooks.py__init__zLocalCLIDebugHook.__init__ s2*DMDO1D D#%D -Dct|jr|jj||y||j|<y)aAdd a tensor filter. See doc of `LocalCLIDebugWrapperSession.add_tensor_filter()` for details. Override default behavior to accommodate the possibility of this method being called prior to the initialization of the underlying `LocalCLIDebugWrapperSession` object. Args: filter_name: See doc of `LocalCLIDebugWrapperSession.add_tensor_filter()` for details. tensor_filter: See doc of `LocalCLIDebugWrapperSession.add_tensor_filter()` for details. N)radd_tensor_filterr)r filter_name tensor_filters rrz#LocalCLIDebugHook.add_tensor_filter<s3   --k=I2?d"";/rcyNrs rbeginzLocalCLIDebugHook.beginQrc x|jstj|j|j|j |j |j|_|jD]+}|jj||j|-|jjtj|jj|jjdd|jj }|jj#|}|j$|_t)j*ddt-j.}|j&tj0j2k(rj|jj5|j6|j8|j:|j<|j>|j@|jB|S|j&tj0jDk(r%|jjG|j6|S)N)rrrr feed_dictoptions) debug_opsnode_name_regex_allowlistop_type_regex_allowlisttensor_dtype_regex_allowlist#tolerate_debug_op_creation_failures)$rrLocalCLIDebugWrapperSessionsessionr rrrrrincrement_run_call_countrOnRunStartRequest original_argsfetchesr'run_call_count on_run_startaction_performed_actionr SessionRunArgsr RunOptionsOnRunStartAction DEBUG_RUN_decorate_run_options_for_debugr( debug_urlsr)r*r+r,r- PROFILE_RUN!_decorate_run_options_for_profile)r run_contextron_run_start_requeston_run_start_responserun_argss r before_runzLocalCLIDebugHook.before_runTs  /KK   --OO!5511 3d55D+ // 55kB DD  224%66!!));+D+D+N+N dD))88:!11>>299D.. j&;&;&=?H !;!;!E!EE ;;     * *)33#==#;;#@@#GG< J$ O   9#=#=#I#I I ==h>N>NO Orctj|j|j}|jj |yr )rOnRunEndRequestr7 run_metadatar on_run_end)rr@ run_valueson_run_end_requests r after_runzLocalCLIDebugHook.after_runs<#2243I3I3=3J3JL$$%78r)readlineNNN) __name__ __module__ __qualname____doc__rrr#rDrKr!rrr r s0 ""& $ .8@* 2h9rr c.eZdZdZ ddZdZdZdZy)DumpingDebugHookz{A debugger hook that dumps debug data to filesystem. Can be used as a hook for `tf.compat.v1.train.MonitoredSession`. Nc<||_||_||_d|_y)aCreate a local debugger command-line interface (CLI) hook. Args: session_root: See doc of `dumping_wrapper.DumpingDebugWrapperSession.__init__`. watch_fn: See doc of `dumping_wrapper.DumpingDebugWrapperSession.__init__`. thread_name_filter: Regular-expression white list for threads on which the wrapper session will be active. See doc of `BaseDebugWrapperSession` for more details. N) _session_root _watch_fnrr)r session_rootwatch_fnrs rrzDumpingDebugHook.__init__s# &DDN1D Drcyr r!r"s rr#zDumpingDebugHook.beginr$rc d}|jsHtj|j|j|j |j |_d}|jj|jj|jj|jj\}}tj}tj||jj ||j"|j$|j&|j(|j*| t-j.dd|}|S)NFrWrT)r=r)r*r+r,r-reset_disk_byte_usager&)rrDumpingDebugWrapperSessionr/rTrUrr0_prepare_run_watch_configr2r3r'rr9r watch_graphgraphr)r*r+r,r-r r8)rr@r[r= watch_options run_optionsrCs rrDzDumpingDebugHook.before_runs!  -HH      >>!55 7d #224!% 5 5 O O!!));+D+D+N+N!PJ '')K!!))"/"I"I - E E%2%O%O  = =3 5 .. k3H Orcyr r!)rr@rIs rrKzDumpingDebugHook.after_runr$rNN)rMrNrOrPrr#rDrKr!rrrRrRs$"&!* B rrRc"eZdZdZ ddZdZy) GrpcDebugHooka A hook that streams debugger-related events to any grpc_debug_server. For example, the debugger data server is a grpc_debug_server. The debugger data server writes debugger-related events it receives via GRPC to logdir. This enables debugging features in Tensorboard such as health pills. When the arguments of debug_utils.watch_graph changes, strongly consider changing arguments here too so that features are available to tflearn users. Can be used as a hook for `tf.compat.v1.train.MonitoredSession`. Ncbd|_||_t|tr|n|g|_||_y)aConstructs a GrpcDebugHook. Args: grpc_debug_server_addresses: (`list` of `str`) A list of the gRPC debug server addresses, in the format of , with or without the "grpc://" prefix. For example: ["localhost:7000", "192.168.0.2:8000"] watch_fn: A function that allows for customizing which ops to watch at which specific steps. See doc of `dumping_wrapper.DumpingDebugWrapperSession.__init__` for details. thread_name_filter: Regular-expression white list for threads on which the wrapper session will be active. See doc of `BaseDebugWrapperSession` for more details. N)_grpc_debug_wrapper_sessionr isinstancelist_grpc_debug_server_addressesrU)rgrpc_debug_server_addressesrWrs rrzGrpcDebugHook.__init__s="(,D$1D 14 8 $ $% % DNrc t|jsFtj|j|j|j |j |_|jj}|jj}|j ||}tj}tj||jj|jj|||j |j"|j$|j&|j(t+j,dd|S)zCalled right before a session is run. Args: run_context: A session_run_hook.SessionRunContext. Encapsulates information on the run. Returns: A session_run_hook.SessionRunArgs object. rZ)r=r)r*r+r,r-Nr&)rgrGrpcDebugWrapperSessionr/rjrUrr2r3r'rr9rr^r_prepare_run_debug_urlsr)r*r+r,r-r r8)rr@r3r'r`ras rrDzGrpcDebugHook.before_runs  + +)5)M)M     + +>>!55 *7d& ''//G))33INN7I6M'')K!!33JJ Y ))"/"I"I - E E%2%O%O  = = @  * * k 33rrc)rMrNrOrPrrDr!rrreres "&4#3rrec2eZdZdZ dfd ZfdZxZS)TensorBoardDebugHookaA tfdbg hook that can be used with TensorBoard Debugger Plugin. This hook is the same as `GrpcDebugHook`, except that it uses a predefined `watch_fn` that 1) uses `DebugIdentity` debug ops with the `gated_grpc` attribute set to `True`, to allow the interactive enabling and disabling of tensor breakpoints. 2) watches all tensors in the graph. This saves the need for the user to define a `watch_fn`. cd}tt| |||||_||_d|_t jy)aConstructor of TensorBoardDebugHook. Args: grpc_debug_server_addresses: gRPC address(es) of debug server(s), as a `str` or a `list` of `str`s. E.g., "localhost:2333", "grpc://localhost:2333", ["192.168.0.7:2333", "192.168.0.8:2333"]. thread_name_filter: Optional filter for thread names. send_traceback_and_source_code: Whether traceback of graph elements and the source code are to be sent to the debug server(s). c4~~tjdgS)NzDebugIdentity(gated_grpc=true))r))r WatchOptions)r3feedss r_gated_grpc_watch_fnz;TensorBoardDebugHook.__init__.._gated_grpc_watch_fn;s 5  # #5688rrZN)superrprrj_send_traceback_and_source_code_sent_graph_versionrregister_signal_handler)rrkrsend_traceback_and_source_coderu __class__s rrzTensorBoardDebugHook.__init__,sO8  .#%-// )DD%+ID(!D((*rc|jrntj|j|jj |j j|j j|j|_ tt|3|Sr ) rxrpublish_tracebackrjr/r_r2r'r3ryrwrprD)rr@r|s rrDzTensorBoardDebugHook.before_runJsr ++!-!?!?  + +[-@-@-F-F  # # - -  # # + +T-E-E"Gd %t 7 DDr)NT)rMrNrOrPrrD __classcell__)r|s@rrprp s! #'.2+<EErrpN)rPtensorflow.core.protobufrtensorflow.python.debug.libr tensorflow.python.debug.wrappersrrrrtensorflow.python.trainingr SessionRunHookr rRrerpr!rrrso#/3<69>7s9(77s9l@ '66@ FJ3$33J3Z0E=0Er