o ?e5@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)LocalCLIDebugHookaCommand-line-interface debugger hook. Can be used as a hook for `tf.compat.v1.train.MonitoredSession`s and `tf.estimator.Estimator`s. Provides a substitute for `tfdbg.LocalCLIDebugWrapperSession` in cases where the session is not directly available. 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_pathrg/home/www/facesmatcher.com/pyenv/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_nameZ tensor_filterrrrr?sz#LocalCLIDebugHook.add_tensor_filtercCdSNrrrrrbeginTzLocalCLIDebugHook.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)$rrZLocalCLIDebugWrapperSessionsessionr r r rrrincrement_run_call_countrZOnRunStartRequest original_argsfetchesr!Zrun_call_countZ on_run_startaction_performed_actionrSessionRunArgsr RunOptionsZOnRunStartActionZ DEBUG_RUNZ_decorate_run_options_for_debugr" debug_urlsr#r$r%r&r'Z PROFILE_RUNZ!_decorate_run_options_for_profile)r run_contextrZon_run_start_requestZon_run_start_responserun_argsrrr before_runWsL      zLocalCLIDebugHook.before_runcCs t|j|j}|j|dSr)rZOnRunEndRequestr-Z run_metadatarZ on_run_end)rr1 run_valuesZon_run_end_requestrrr after_runszLocalCLIDebugHook.after_run)r NNN) __name__ __module__ __qualname____doc__rrrr3r5rrrrr s   4r c@s6eZdZdZ  d ddZddZddZd d ZdS) DumpingDebugHookzA debugger hook that dumps debug data to filesystem. Can be used as a hook for `tf.compat.v1.train.MonitoredSession`s and `tf.estimator.Estimator`s. 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)rZ session_rootwatch_fnrrrrrs zDumpingDebugHook.__init__cCrrrrrrrrrzDumpingDebugHook.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)NFr=rT)r0r#r$r%r&r'reset_disk_byte_usager )rrZDumpingDebugWrapperSessionr(r;r<r r)Z_prepare_run_watch_configr*r+r!rr/r watch_graphgraphr#r$r%r&r'rr.)rr1r?r0 watch_options run_optionsr2rrrr3s:   zDumpingDebugHook.before_runcCrrr)rr1r4rrrr5rzDumpingDebugHook.after_runNN)r6r7r8r9rrr3r5rrrrr:s  !r:c@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`s and `tf.estimator.Estimator`s. 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_addressesr<)rgrpc_debug_server_addressesr=rrrrrs 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. r>)r0r#r$r%r&r'Nr )rFrZGrpcDebugWrapperSessionr(rIr<r r*r+r!rr/rr@rAZprepare_run_debug_urlsr#r$r%r&r'rr.)rr1r+r!rBrCrrrr3s4   zGrpcDebugHook.before_runrD)r6r7r8r9rr3rrrrrEs  rEcs2eZdZdZ  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#)rZ WatchOptions)r+Zfeedsrrr_gated_grpc_watch_fn@sz;TensorBoardDebugHook.__init__.._gated_grpc_watch_fnr>N)superrKrrI_send_traceback_and_source_code_sent_graph_versionrZregister_signal_handler)rrJrZsend_traceback_and_source_coderL __class__rrr1s  zTensorBoardDebugHook.__init__cs:|jrt|j|jj|jj|jj|j |_ t t | |Sr) rOrZpublish_tracebackrIr(rAr*r!r+rPrNrKr3)rr1rQrrr3Os  zTensorBoardDebugHook.before_run)NT)r6r7r8r9rr3 __classcell__rrrQrrK%s  rKN)r9Ztensorflow.core.protobufrZtensorflow.python.debug.librZ tensorflow.python.debug.wrappersrrrrZtensorflow.python.trainingrZSessionRunHookr r:rErKrrrrs       yDN