B һd7@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@s:eZdZdZdddZddZdd 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. cursesNcCs(||_||_||_d|_i|_||_dS)aCreate a local debugger command-line interface (CLI) hook. Args: ui_type: (`str`) requested user-interface type. Currently supported: (curses | 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_pathrX/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/debug/wrappers/hooks.py__init__#s zLocalCLIDebugHook.__init__cCs$|jr|j||n ||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_filtercCsdS)Nr)rrrrbeginTszLocalCLIDebugHook.beginc Cs|jsJtj|j|j|j|j|jd|_x"|jD]}|j ||j|q.W|j t |j j|j jdd|jj}|j|}|j|_tjddtd}|jt jjkr|jj|j|j|j|j|j|j |j!dn|jt jj"kr|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_argsfetchesrZrun_call_countZ on_run_startaction_performed_actionrSessionRunArgsr RunOptionsZOnRunStartActionZ DEBUG_RUNZ_decorate_run_options_for_debugr debug_urlsrrr r!r"Z PROFILE_RUNZ!_decorate_run_options_for_profile)r run_contextrZon_run_start_requestZon_run_start_responserun_argsrrr before_runWs>     zLocalCLIDebugHook.before_runcCs t|j|j}|j|dS)N)rZOnRunEndRequestr(Z run_metadatarZ on_run_end)rr, run_valuesZon_run_end_requestrrr after_runszLocalCLIDebugHook.after_run)r NNN) __name__ __module__ __qualname____doc__rrrr.r0rrrrr s 4r c@s2eZdZdZd ddZddZdd Zd 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. NTcCs"||_||_||_||_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. log_usage: (bool) Whether usage is to be logged. N) _session_root _watch_fnr _log_usager)rZ session_rootwatch_fnr log_usagerrrrs zDumpingDebugHook.__init__cCsdS)Nr)rrrrrszDumpingDebugHook.beginc Csd}|js.tj|j|j|j|j|jd|_d}|j|j |j j |j j \}}t }tj||jj||j|j|j|j|j|d tjdd|d}|S)NF)r9rr:T)r+rrr r!r"reset_disk_byte_usage)rr)rrZDumpingDebugWrapperSessionr#r6r7r r8r$Z_prepare_run_watch_configr%r&rrr*r watch_graphgraphrrr r!r"rr))rr,r;r+ watch_options run_optionsr-rrrr.s4   zDumpingDebugHook.before_runcCsdS)Nr)rr,r/rrrr0szDumpingDebugHook.after_run)NNT)r1r2r3r4rrr.r0rrrrr5s "r5c@s"eZdZdZdddZddZdS) 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. NTcCs2d|_||_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. log_usage: (bool) Whether usage is to be logged. N)_grpc_debug_wrapper_sessionr isinstancelist_grpc_debug_server_addressesr7r8)rgrpc_debug_server_addressesr9rr:rrrrs zGrpcDebugHook.__init__c Cs|js&tj|j|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. )r9rr:)r+rrr r!r"N)rr)rArZGrpcDebugWrapperSessionr#rDr7r r8r%r&rrr*rr<r=Zprepare_run_debug_urlsrrr r!r"rr))rr,r&rr>r?rrrr.s.    zGrpcDebugHook.before_run)NNT)r1r2r3r4rr.rrrrr@s   r@cs.eZdZdZdfdd 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)a,Constructor 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). log_usage: Whether the usage of this class is to be logged (if applicable). cSs~~tjdgdS)NzDebugIdentity(gated_grpc=true))r)rZ WatchOptions)r&Zfeedsrrr_gated_grpc_watch_fnKsz;TensorBoardDebugHook.__init__.._gated_grpc_watch_fn)r9rr:N)superrFrrD_send_traceback_and_source_code_sent_graph_versionrZregister_signal_handler)rrErZsend_traceback_and_source_coder:rG) __class__rrr9s zTensorBoardDebugHook.__init__cs:|jr*t|j|jj|jj|jj|j |_ t t | |S)N) rJrZpublish_tracebackrDr#r=r%rr&rKrIrFr.)rr,)rLrrr.[s  zTensorBoardDebugHook.before_run)NTT)r1r2r3r4rr. __classcell__rr)rLrrF-s  rFN)r4Ztensorflow.core.protobufrZtensorflow.python.debug.librZ tensorflow.python.debug.wrappersrrrrZtensorflow.python.trainingrZSessionRunHookr r5r@rFrrrrs       yHR