o ?e@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) frameworkcCsDddlm}|j|kr t||}|j||t|dd|jS|S)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_source)tensorflow.python.debug.librversionrZ get_run_keyZsend_graph_tracebacks traceback extract_stack)Zdebug_server_urlsgraph feed_dictfetchesZold_graph_versionrZrun_keyr n/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/debug/wrappers/grpc_wrapper.pypublish_tracebacks    rc@s.eZdZdZ  d ddZddZddZdS) GrpcDebugWrapperSessionz=Debug Session wrapper that send debug data to gRPC stream(s).NcCstjj||||dt|tr||g|_dSt|tr=g|_|D]}t|ts1tdt ||j ||q"dStdt |)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. 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_addressesrraddressr r r rAs,   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_urlsmsz.GrpcDebugWrapperSession.prepare_run_debug_urlscCs|tjs tj|S|SN) startswithrZGRPC_URL_PREFIX)rr r r r r~s  z+GrpcDebugWrapperSession._normalize_grpc_url)NN)__name__ __module__ __qualname____doc__rr!rr r r r r>s , rcCs0 td}|dvrtdn|dvrdSq)NTz' SIGINT received. Quit program? (Y/n): )Yyr)Nn)inputstripsysexit)Z unused_signalZ unused_frameresponser r r _signal_handlers  r2cCs*z ttjtWdStyYdSwr")signalSIGINTr2 ValueErrorr r r r register_signal_handlers  r6cs@eZdZdZ  dfdd Z      d 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`. NTcs2dd}tj||||d||_d|_tdS)aConstructor 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). cSs~~tjdgdS)NzDebugIdentity(gated_grpc=true))Z debug_ops)rZ WatchOptions)r Zfeedsr r r _gated_grpc_watch_fnszETensorBoardDebugWrapperSession.__init__.._gated_grpc_watch_fnrN)superr_send_traceback_and_source_code_sent_graph_versionr6)rrrrZsend_traceback_and_source_coder8 __class__r r rs z'TensorBoardDebugWrapperSession.__init__c s8|jrt|j|j|||j|_tj|||||||dS)N)r options run_metadatacallable_runnercallable_runner_argscallable_options)r;rrr r<r:run)rr r r?r@rArBrCr=r r rDs z"TensorBoardDebugWrapperSession.run)NT)NNNNNN)r$r%r&r'rrD __classcell__r r r=r r7s$r7)r'r3r/rrrZ tensorflow.python.debug.wrappersrrrrr2r6r7r r r r s  &E