BVh9dZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdaej*Zdad ZGd d eZGddeZeddddZedddZedddZedddZedddZ eddGdde!Z"y)a!TensorFlow 2.0 Profiler for both Eager Mode and Graph Mode. The profiler has two mode: - Programmatic Mode: start(), stop() and Profiler class. It will perform when calling start() or create Profiler class and will stop when calling stop() or destroying Profiler class. - On-demand Mode: start_profiler_server(). It will perform profiling when receive profiling request. NOTE: Only one active profiler session is allowed. Use of simultaneous Programmatic Mode and On-demand Mode is undefined and will likely fail. NOTE: The Keras TensorBoard callback will automatically perform sampled profiling. Before enabling customized profiling, set the callback flag "profile_batches=[]" to disable automatic sampled profiling. customized profiling. N)_pywrap_events_writer)context)errors)gfile) tf_logging)_pywrap_profiler)compat) deprecatedz.profile-emptyc eZdZy)ProfilerAlreadyRunningErrorN__name__ __module__ __qualname__P/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/eager/profiler.pyr r 6rr c eZdZy)ProfilerNotRunningErrorNr rrrrr:rrrz 2020-07-01z-use `tf.profiler.experimental.start` instead.ct5t tdtjtj k(rtj tja tjd||ni dddy#tj$r!tjdtdwxYw#1swYyxYw)zStart profiling. Args: options: profiler options. Raises: ProfilerAlreadyRunningError: If another profiling session is running. NzAnother profiler is running.zAnother profiler session is running which is probably created by profiler server. Please avoid using profiler server and profiler APIs at the same time.)_profiler_lock _profilerr rdefault_execution_mode EAGER_MODEensure_initializedrProfilerSessionstartrAlreadyExistsErrorloggingwarning)optionss rrr>s H '(F GG%%););;   " 002IHoobW%8'bA H H  $ $H ooCD ((F GG H H HsAB>#B4B;;B>>Cz,use `tf.profiler.experimental.stop` instead.c8t5t tdtjtj k(r,tjj jtj}datdz a ddd|S#1swYSxYw)a Stop current profiling session and return its result. Returns: A binary string of tensorflow.tpu.Trace. User can write the string to file for offline analysis by tensorboard. Raises: ProfilerNotRunningError: If there is no active profiling session. Nz.Cannot stop profiling. No profiler is running.) rrrrrrexecutorwaitstop_run_num)results rr(r(Xs # : <<%%););; oo  %%' ^^ FI MH - -s A>BBzE`tf.python.eager.profiler` has deprecated, use `tf.profiler` instead.cBtj|D]}|jtsyt j t jtjj|d}|jt jty)zCreate an empty event file if not already exists. This event file indicates that we have a plugins/profile/ directory in the current logdir. Args: logdir: log directory. Nevents) r ListDirectoryendswith_EVENT_FILE_SUFFIXr EventsWriterr as_bytesospathjoinInitWithSuffix)logdir file_name event_writers rmaybe_create_event_filer9qsu&&v. i,-  '33 oobggll68457,foo.@ABrctjj|ddtjj j d}t j|t|t jtjj|dd5}|j|dddy#1swYyxYw)zSave profile result to TensorBoard logdir. Args: logdir: log directory read by TensorBoard. result: profiling result returned by stop(). pluginsprofilez%Y-%m-%d_%H-%M-%Sz local.tracewbN) r2r3r4datetimenowstrftimerMakeDirsr9Openwrite)r6r* plugin_dirfs rsaverFsww|| i&&':;=*..&! zz"'',,z=94@AGGFOs "B==Cz,use `tf.profiler.experimental.server.start`.ctjtjk(rtjt j |y)aStart a profiler grpc server that listens to given port. The profiler server will keep the program running even the training finishes. Please shutdown the server with CTRL-C. It can be used in both eager mode and graph mode. The service defined in tensorflow/core/profiler/profiler_service.proto. Please use tensorflow/contrib/tpu/profiler/capture_tpu_profile to capture tracable file following https://cloud.google.com/tpu/docs/cloud-tpu-tools#capture_trace Args: port: port profiler server listens to. N)rrrrr start_server)ports rstart_profiler_serverrJs2 ##w'9'99  %rz/use `tf.profiler.experimental.Profile` instead.c"eZdZdZdZdZdZy)ProfilerzContext-manager eager profiler api. Example usage: ```python with Profiler("/path/to/logdir"): # do some work ``` c||_yN)_logdir)selfr6s r__init__zProfiler.__init__s DLrctyrN)r)rPs r __enter__zProfiler.__enter__s GrcDt}t|j|yrN)r(rFrO)rPtypvaluetbr*s r__exit__zProfiler.__exit__s VFvrN)rrr__doc__rQrSrXrrrrLrLs rrLrN)#rYr>r2 threadingtensorflow.python.clientrtensorflow.python.eagerrtensorflow.python.frameworkrtensorflow.python.platformrrr!#tensorflow.python.profiler.internalrtensorflow.python.utilr "tensorflow.python.util.deprecationr rLockrr)r/ Exceptionr rrr(r9rFrJobjectrLrrrres"$ :+.,<@)9  ! &)i LIJHKH2 LHIJ0 KC C$ K     LHI&J&$ LKLvMr