B һd!@srdZddlZddlZddlZddlmZddlmZddZGdddej Z d d Z d d Z Gd dde Z dS)z?Debugger wrapper session that sends debug data to file:// URLs.N)common) frameworkcCsHddlm}|j|kr@t||}|j||t|dd|jS|SdS)aPublish traceback and source code if graph version is new. `graph.version` is compared with `old_graph_version`. If the former is higher (i.e., newer), the graph traceback and the associated source code is sent to the debug server at the specified gRPC URLs. Args: debug_server_urls: A single gRPC debug server URL as a `str` or a `list` of debug server URLs. graph: A Python `tf.Graph` object. feed_dict: Feed dictionary given to the `Session.run()` call. fetches: Fetches from the `Session.run()` call. old_graph_version: Old graph version to compare to. Returns: If `graph.version > old_graph_version`, the new graph version as an `int`. Else, the `old_graph_version` is returned. r) source_remoteT)Z send_sourceN)tensorflow.python.debug.librversionrZ get_run_keyZsend_graph_tracebacks traceback extract_stack)Zdebug_server_urlsgraph feed_dictfetchesZold_graph_versionrZrun_keyr _/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/debug/wrappers/grpc_wrapper.pypublish_tracebacks    rc@s*eZdZdZd ddZddZdd ZdS) GrpcDebugWrapperSessionz=Debug Session wrapper that send debug data to gRPC stream(s).NTcCs|rtjj||||dt|tr2||g|_n\t|tr~g|_xJ|D]0}t|tsftdt ||j ||qHWntdt |dS)aConstructor of DumpingDebugWrapperSession. Args: sess: The TensorFlow `Session` object being wrapped. grpc_debug_server_addresses: (`str` or `list` of `str`) Single or a list of the gRPC debug server addresses, in the format of , with or without the "grpc://" prefix. For example: "localhost:7000", ["localhost:7000", "192.168.0.2:8000"] watch_fn: (`Callable`) A Callable that can be used to define per-run debug ops and watched tensors. See the doc of `NonInteractiveDebugWrapperSession.__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 the usage of this class is to be logged. Raises: TypeError: If `grpc_debug_server_addresses` is not a `str` or a `list` of `str`. )watch_fnthread_name_filterzGExpected type str in list grpc_debug_server_addresses, received type %szJExpected type str or list in grpc_debug_server_addresses, received type %sN) r!NonInteractiveDebugWrapperSession__init__ isinstancestr_normalize_grpc_url_grpc_debug_server_urlslist TypeErrortypeappend)selfsessgrpc_debug_server_addressesrr log_usageaddressr r r rBs      z GrpcDebugWrapperSession.__init__cCs|jS)aImplementation of abstract method in superclass. See doc of `NonInteractiveDebugWrapperSession.prepare_run_debug_urls()` for details. Args: fetches: Same as the `fetches` argument to `Session.run()` feed_dict: Same as the `feed_dict` argument to `Session.run()` Returns: debug_urls: (`str` or `list` of `str`) file:// debug URLs to be used in this `Session.run()` call. )r)rr r r r r prepare_run_debug_urlstsz.GrpcDebugWrapperSession.prepare_run_debug_urlscCs|tjstj|S|S)N) startswithrZGRPC_URL_PREFIX)rr r r r rsz+GrpcDebugWrapperSession._normalize_grpc_url)NNT)__name__ __module__ __qualname____doc__rr!rr r r r r?s  -rcCs4x.td}|dkr"tdq|dkrPqWdS)Nz' SIGINT received. Quit program? (Y/n): )Yyr)Nn)inputstripsysexit)Z unused_signalZ unused_frameresponser r r _signal_handlers   r1cCs,yttjtWntk r&YnXdS)N)signalSIGINTr1 ValueErrorr r r r register_signal_handlersr5cs0eZdZdZdfdd Zd fdd ZZS) TensorBoardDebugWrapperSessionaA tfdbg Session wrapper that can be used with TensorBoard Debugger Plugin. This wrapper is the same as `GrpcDebugWrapperSession`, 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`. NTcs4dd}tj|||||d||_d|_tdS)auConstructor of TensorBoardDebugWrapperSession. Args: sess: The `tf.compat.v1.Session` instance to be wrapped. 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))Z debug_ops)rZ WatchOptions)r Zfeedsr r r _gated_grpc_watch_fnszETensorBoardDebugWrapperSession.__init__.._gated_grpc_watch_fn)rrrN)superr_send_traceback_and_source_code_sent_graph_versionr5)rrrrZsend_traceback_and_source_coderr7) __class__r r rsz'TensorBoardDebugWrapperSession.__init__c s8|jrt|j|j|||j|_tj|||||||dS)N)r options run_metadatacallable_runnercallable_runner_argscallable_options)r:rrr r;r9run)rr r r=r>r?r@rA)r<r r rBs  z"TensorBoardDebugWrapperSession.run)NTT)NNNNNN)r#r$r%r&rrB __classcell__r r )r<r r6s #r6)r&r2r.rrrZ tensorflow.python.debug.wrappersrrrrr1r5r6r r r r s  &K