B d@stdZddlZddlZddlZddlZddlmZeZd ddZ Gdddej Z e Z ej dd Zd d ZdS) z%Utilities for measuring elapsed time.N) tb_loggingcCs\|dkrtj}t|tr&|}t||S|}t|dd}|dkrFt|}t||}||SdS)aLog latency in a function or region. Three usages are supported. As a decorator: >>> @log_latency ... def function_1(): ... pass ... As a decorator with a custom label for the region: >>> @log_latency("custom_label") ... def function_2(): ... pass ... As a context manager: >>> def function_3(): ... with log_latency("region_within_function"): ... pass ... Args: region_name_or_function_to_decorate: Either: a `str`, in which case the result of this function may be used as either a decorator or a context manager; or a callable, in which case the result of this function is a decorated version of that callable. log_level: Optional integer logging level constant. Defaults to `logging.INFO`. Returns: A decorated version of the input callable, or a dual decorator/context manager with the input region name. N __qualname__)loggingINFO isinstancestr _log_latencygetattr)Z#region_name_or_function_to_decorate log_levelZ region_nameZfunction_to_decoratequalname decoratorr I/var/www/html/venv/lib/python3.7/site-packages/tensorboard/util/timing.py log_latencys'    rc@seZdZddZdS)_ThreadLocalStorecCs d|_dS)Nr) nesting_level)selfr r r__init__Rsz_ThreadLocalStore.__init__N)__name__ __module__rrr r r rrQsrc cst|sdVdStj}zLt}|dt_d|}t}d|j|j|f}t |d||dVWd|t_t|}t |d|||XdS)Nz z%s[%x]%sz %s ENTER %sz%s LEAVE %s - %0.6fs elapsed) logger isEnabledFor_storertime threadingcurrent_threadnameident_log)rr Z start_levelstartedindentthreadprefixelapsedr r rrYs(    rcGstj||f|dS)N)rlog)r msgargsr r rrtsr)N)__doc__ contextlibrrrZtensorboard.utilrZ get_loggerrrlocalrrcontextmanagerrrr r r rs  6