BVhdZddlZddlZddlZddlmZddlmZdZGddejZ dZ d Z Gd d e Z y) z?Debugger wrapper session that sends debug data to file:// URLs.N)common) frameworkcddlm}|j|kDrJtj||}|j ||t j|d|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) send_source)tensorflow.python.debug.librversionr get_run_keysend_graph_tracebacks traceback extract_stack)debug_server_urlsgraph feed_dictfetchesold_graph_versionrrun_keys ]/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/debug/wrappers/grpc_wrapper.pypublish_tracebackrs`48 ]]&&  G4G''7I$;$;$=u( == c(eZdZdZ ddZdZdZy)GrpcDebugWrapperSessionz=Debug Session wrapper that send debug data to gRPC stream(s).Nctjj||||t|tr|j |g|_yt|tr`g|_|D]S}t|tstdt|z|j j|j |Uytdt|z)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_addressesrraddresss rrz GrpcDebugWrapperSession.__init__As2//88 dX:L9N-s3  " "#> ?&Ad" / 6%'d"0O''3'!#'=12 2 $$++D,D,DW,MN O  #$?@ A BBrc|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")r'rrs rprepare_run_debug_urlsz.GrpcDebugWrapperSession.prepare_run_debug_urlsms  ' ''rcj|jtjstj|zS|SN) startswithrGRPC_URL_PREFIX)r'r*s rr!z+GrpcDebugWrapperSession._normalize_grpc_url~s5%%f&<&<=  " "W ,LCJLr)NN)__name__ __module__ __qualname____doc__rr,r!rrrr>sE "& *BX("Lrrcv tdj}|dvrtjdn|dvry9)Nz' SIGINT received. Quit program? (Y/n): )Yyr)Nn)inputstripsysexit) unused_signal unused_frameresponses r_signal_handlerrCs< ?@FFHH>! hhqk Z  rct tjtjty#t$rYywxYwr.)signalSIGINTrC ValueErrorr5rrregister_signal_handlerrHs+  MM&--1   s (+ 77c@eZdZdZ dfd Z dfd ZxZS)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`. cbd}t|||||||_d|_t y)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). c4~~tjdgS)NzDebugIdentity(gated_grpc=true)) debug_ops)r WatchOptions)rfeedss r_gated_grpc_watch_fnzETensorBoardDebugWrapperSession.__init__.._gated_grpc_watch_fns 5  # #5688rrN)superr_send_traceback_and_source_code_sent_graph_versionrH)r'r(r)rsend_traceback_and_source_coderP __class__s rrz'TensorBoardDebugWrapperSession.__init__sC 8  G #%- / ,JD( "Drc |jr2t|j|j|||j|_t ||||||||S)N)roptions run_metadatacallable_runnercallable_runner_argscallable_options)rSrr"rrTrRrun) r'rrrXrYrZr[r\rVs rr]z"TensorBoardDebugWrapperSession.runsd ++!2  & & Iw  " ""$d 7;!'1)  ++r)NT)NNNNNN)r1r2r3r4rr] __classcell__)rVs@rrJrJs4 #'.2 H#++rrJ)r4rEr>r rr tensorflow.python.debug.wrappersrrrrrCrHrJr5rrr`sPF .6#LBLiIIBLJ  A+%<A+r