o 2h4@sdZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z Gd d d e j Z Gd d d e j ZGd dde j ZGdddeZdS)ztfdbg CLI as SessionRunHook.) config_pb2) debug_utils)dumping_wrapper) framework) grpc_wrapper)local_cli_wrapper)session_run_hookc@sBeZdZdZ    dddZddZdd Zd d Zd d ZdS)LocalCLIDebugHookzlCommand-line-interface debugger hook. Can be used as a hook for `tf.compat.v1.train.MonitoredSession`. readlineNcCs(||_||_||_d|_i|_||_dS)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_pathra/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/debug/wrappers/hooks.py__init__ s  zLocalCLIDebugHook.__init__cCs&|jr |j||dS||j|<dS)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_filterrrrr<sz#LocalCLIDebugHook.add_tensor_filtercCdSNrrrrrbeginQzLocalCLIDebugHook.beginc Cs|js#tj|j|j|j|j|jd|_|jD] }|j ||j|q|j t |j j|j jdd|jj}|j|}|j|_tjddtd}|jt jjkrh|jj|j|j|j|j|j|j |j!d|S|jt jj"krv|j#|j|S)N)rrrr feed_dictoptions) debug_opsnode_name_regex_allowlistop_type_regex_allowlisttensor_dtype_regex_allowlist#tolerate_debug_op_creation_failures)$rrLocalCLIDebugWrapperSessionsessionr r r rrrincrement_run_call_countrOnRunStartRequest original_argsfetchesr"run_call_count on_run_startaction_performed_actionrSessionRunArgsr 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_argsrrr before_runTsL      zLocalCLIDebugHook.before_runcCs t|j|j}|j|dSr)rOnRunEndRequestr2 run_metadatar on_run_end)rr; run_valueson_run_end_requestrrr after_runszLocalCLIDebugHook.after_run)r NNN) __name__ __module__ __qualname____doc__rrrr?rErrrrr s  4r c@s6eZdZdZ  d ddZddZddZd d ZdS) DumpingDebugHookz{A debugger hook that dumps debug data to filesystem. Can be used as a hook for `tf.compat.v1.train.MonitoredSession`. NcCs||_||_||_d|_dS)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_fnr r)r session_rootwatch_fnrrrrrs zDumpingDebugHook.__init__cCrrrrrrrrr zDumpingDebugHook.beginc Csd}|jstj|j|j|j|jd|_d}|j|j|j j |j j \}}t }tj||jj||j|j|j|j|j|d tjdd|d}|S)NFrNrT)r8r$r%r&r'r(reset_disk_byte_usager!)rrDumpingDebugWrapperSessionr*rKrLr r+_prepare_run_watch_configr-r.r"rr4r watch_graphgraphr$r%r&r'r(rr3)rr;rPr8 watch_options run_optionsr>rrrr?s:   zDumpingDebugHook.before_runcCrrr)rr;rCrrrrEr zDumpingDebugHook.after_runNN)rFrGrHrIrrr?rErrrrrJs  !rJc@s&eZdZdZ  dddZddZdS) 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`. NcCs,d|_||_t|tr |n|g|_||_dS)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_addressesrL)rgrpc_debug_server_addressesrNrrrrrs zGrpcDebugHook.__init__c Cs|jstj|j|j|j|jd|_|jj}|jj }|||}t }t j ||jj|j|||j|j|j|j|jdtjdd|dS)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. rO)r8r$r%r&r'r(Nr!)rYrGrpcDebugWrapperSessionr*r\rLr r-r.r"rr4rrSrTprepare_run_debug_urlsr$r%r&r'r(rr3)rr;r.r"rUrVrrrr?s4   zGrpcDebugHook.before_runrW)rFrGrHrIrr?rrrrrXs  rXcs2eZdZdZ  dfdd ZfddZZS) 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`. NTcs<dd}tt|j|||d||_||_d|_tdS)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). cSs~~tjdgdS)NzDebugIdentity(gated_grpc=true))r$)r WatchOptions)r.feedsrrr_gated_grpc_watch_fn;sz;TensorBoardDebugHook.__init__.._gated_grpc_watch_fnrON)superr`rr\_send_traceback_and_source_code_sent_graph_versionrregister_signal_handler)rr]rsend_traceback_and_source_coderc __class__rrr,s  zTensorBoardDebugHook.__init__cs:|jrt|j|jj|jj|jj|j |_ t t | |Sr) rfrpublish_tracebackr\r*rTr-r"r.rgrer`r?)rr;rjrrr?Js  zTensorBoardDebugHook.before_run)NT)rFrGrHrIrr? __classcell__rrrjrr` s  r`N)rItensorflow.core.protobufrtensorflow.python.debug.libr tensorflow.python.debug.wrappersrrrrtensorflow.python.trainingrSessionRunHookr rJrXr`rrrrs       vCM