o ?e'A @sdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZed d Zeejejejd eejejejd eejejejd gZeejejejd eejej ej!d eej"ej#ej$d gZ%eej&ej'ej(d eej)ej*ej+d eej,ej-ej.d eej/ej0ej1d eej2ej3ej4d gZ5eej6ej7ej8d eej9ej:ej;d eejd gZ?eej@ejAejBd eejCejDejEd eejFejGejHd gZIGd d d eJZKGdddeJZLGdddeKZMGdddeJZNGdddeKZOGdddeJZPGdddeKZQGdddeJZRedgdGdddeKZSGd d!d!eJZTGd"d#d#eJZUGd$d%d%eUZVGd&d'd'eKZWgZXGd(d)d)eJZYd*d+ZZdS),zTensorFlow monitoring APIs.N) summary_pb2) pywrap_tfe)pywrap_tf_session) c_api_util)compat) tf_exportZ MetricMethodzcreate delete get_cell)createdeleteget_cellc@s0eZdZdZgdZddZddZddZd S) MetriczThe base class of metric.)_metric _metric_name_metric_methods _label_lengthcGsP||_||_||_|t|jkrtd|jt|j|j|jj||_dS)zCreates a new metric. Args: metric_name: name of the metric class. metric_methods: list of swig metric methods. label_length: length of label args. *args: the arguments to call create method. z(Cannot create {} metric with label >= {}N)r rrlen ValueErrorformatrr )selfZ metric_nameZmetric_methodsZ label_lengthargsrc/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/eager/monitoring.py__init__rs  zMetric.__init__cCsFz |j|jj}|j}Wn tyYdSw|dur!||dSdSN)rrr r AttributeError)rdeleterZmetricrrr__del__s   zMetric.__del__cGs>t||jkrtd|j|j|j|jj|jg|RS)Retrieves the cell.zThe {} expects taking {} labels)rrrrr rr r rlabelsrrrr s zMetric.get_cellN)__name__ __module__ __qualname____doc__ __slots__rrr rrrrr ms  r c@.eZdZdZdgZddZddZddZd S) CounterCellz+CounterCell stores each value of a Counter._cellcC ||_dS)z`Creates a new CounterCell. Args: cell: A c pointer of TFE_MonitoringCounterCell. Nr&rcellrrrr zCounterCell.__init__cCt|j|dS)zQAtomically increments the value. Args: value: non-negative value. N)rZ$TFE_MonitoringCounterCellIncrementByr&rvaluerrr increase_byzCounterCell.increase_bycC t|jSRetrieves the current value.)rZTFE_MonitoringCounterCellValuer&rrrrr. zCounterCell.valueN)rr r!r"r#rr/r.rrrrr%  r%c0eZdZdZgZfddZfddZZS)CounterzA stateful class for updating a cumulative integer metric. This class encapsulates a set of values (or a single value for a label-less metric). Each value is identified by a tuple of labels. The class allows the user to increment each value. c(tt|jdtt|||g|RdS)zCreates a new Counter. Args: name: name of the new metric. description: description of the new metric. *labels: The label list of the new metric. r8N)superr8r_counter_methodsrrname descriptionr __class__rrr  zCounter.__init__cttt|j|Sr)r%r:r8r rr?rrr zCounter.get_cellrr r!r"r#rr __classcell__rrr?rr8   r8c@r$) IntGaugeCellz/A single integer value stored in an `IntGauge`.r&cCr')zbCreates a new IntGaugeCell. Args: cell: A c pointer of TFE_MonitoringIntGaugeCell. Nr(r)rrrrr+zIntGaugeCell.__init__cCr,)zEAtomically set the value. Args: value: integer value. N)rZTFE_MonitoringIntGaugeCellSetr&r-rrrsetr0zIntGaugeCell.setcCr1r2)rZTFE_MonitoringIntGaugeCellValuer&r4rrrr.r5zIntGaugeCell.valueNrr r!r"r#rrIr.rrrrrHr6rHcr7)IntGaugezA stateful class for updating a gauge-like integer metric. This class encapsulates a set of integer values (or a single value for a label-less metric). Each value is identified by a tuple of labels. The class allows the user to set each value. cr9)zCreates a new IntGauge. Args: name: name of the new metric. description: description of the new metric. *labels: The label list of the new metric. rKN)r:rKr_int_gauge_methodsrr<r?rrrrAzIntGauge.__init__crBrC)rHr:rKr rr?rrr rDzIntGauge.get_cellrErrr?rrKrGrKc@r$) StringGaugeCellz1A single string value stored in an `StringGauge`.r&cCr')zhCreates a new StringGaugeCell. Args: cell: A c pointer of TFE_MonitoringStringGaugeCell. Nr(r)rrrrr+zStringGaugeCell.__init__cCr,)zDAtomically set the value. Args: value: string value. N)rZ TFE_MonitoringStringGaugeCellSetr&r-rrrrI r0zStringGaugeCell.setcCsLt}t|j|t|d}Wd|S1swY|S)r3zutf-8N)r tf_bufferrZ"TFE_MonitoringStringGaugeCellValuer&r TF_GetBufferdecode)rbuffer_r.rrrr.s  zStringGaugeCell.valueNrJrrrrrMr6rMcr7) StringGaugezA stateful class for updating a gauge-like string metric. This class encapsulates a set of string values (or a single value for a label-less metric). Each value is identified by a tuple of labels. The class allows the user to set each value. cr9)zCreates a new StringGauge. Args: name: name of the new metric. description: description of the new metric. *labels: The label list of the new metric. rRN)r:rRr_string_gauge_methodsrr<r?rrr'   zStringGauge.__init__crBrC)rMr:rRr rr?rrr 2rDzStringGauge.get_cellrErrr?rrRrGrRc@r$) BoolGaugeCellz0A single boolean value stored in an `BoolGauge`.r&cCr')zdCreates a new BoolGaugeCell. Args: cell: A c pointer of TFE_MonitoringBoolGaugeCell. Nr(r)rrrr<r+zBoolGaugeCell.__init__cCr,)zBAtomically set the value. Args: value: bool value. N)rZTFE_MonitoringBoolGaugeCellSetr&r-rrrrIDr0zBoolGaugeCell.setcCr1r2)rZ TFE_MonitoringBoolGaugeCellValuer&r4rrrr.Lr5zBoolGaugeCell.valueNrJrrrrrU7r6rUz!__internal__.monitoring.BoolGauge)v1cr7) BoolGaugezA stateful class for updating a gauge-like bool metric. This class encapsulates a set of boolean values (or a single value for a label-less metric). Each value is identified by a tuple of labels. The class allows the user to set each value. cr9)zCreates a new BoolGauge. Args: name: name of the new metric. description: description of the new metric. *labels: The label list of the new metric. rWN)r:rWr_bool_gauge_methodsrr<r?rrr\rTzBoolGauge.__init__crBrC)rUr:rWr rr?rrr grDzBoolGauge.get_cellrErrr?rrWQs   rWc@r$) SamplerCellz+SamplerCell stores each value of a Sampler.r&cCr')z`Creates a new SamplerCell. Args: cell: A c pointer of TFE_MonitoringSamplerCell. Nr(r)rrrrqr+zSamplerCell.__init__cCr,)zBAtomically add a sample. Args: value: float value. N)rZTFE_MonitoringSamplerCellAddr&r-rrraddyr0zSamplerCell.addcCs\t}t|j|t|}Wdn1swYt}| t ||S)zRetrieves the current distribution of samples. Returns: A HistogramProto describing the distribution of samples. N) rrNrZTFE_MonitoringSamplerCellValuer&rrOrZHistogramProtoZParseFromStringras_bytes)rrQZ proto_dataZhistogram_protorrrr.s  zSamplerCell.valueN)rr r!r"r#rrZr.rrrrrYlr6rYc@s&eZdZdZdgZddZddZdS)Bucketsz&Bucketing strategies for the samplers.bucketscCr')z[Creates a new Buckets. Args: buckets: A c pointer of TFE_MonitoringBuckets. N)r])rr]rrrrr+zBuckets.__init__cCst|jdSr)rZTFE_MonitoringDeleteBucketsr]r4rrrrszBuckets.__del__N)rr r!r"r#rrrrrrr\s  r\cs$eZdZdZgZfddZZS)ExponentialBucketszExponential bucketing strategy. Sets up buckets of the form: [-DBL_MAX, ..., scale * growth^i, scale * growth_factor^(i + 1), ..., DBL_MAX]. cstt|t|||dS)z|Creates a new exponential Buckets. Args: scale: float growth_factor: float bucket_count: integer N)r:r^rrZ#TFE_MonitoringNewExponentialBuckets)rscaleZ growth_factorZ bucket_countr?rrrs zExponentialBuckets.__init__)rr r!r"r#rrFrrr?rr^sr^cr7)SampleraRA stateful class for updating a cumulative histogram metric. This class encapsulates a set of histograms (or a single histogram for a label-less metric) configured with a list of increasing bucket boundaries. Each histogram is identified by a tuple of labels. The class allows the user to add a sample to each histogram value. cs,tt|jdtt|||j|g|RdS)zCreates a new Sampler. Args: name: name of the new metric. buckets: bucketing strategy of the new metric. description: description of the new metric. *labels: The label list of the new metric. r`N)r:r`r_sampler_methodsrr])rr=r]r>rr?rrrs   zSampler.__init__crBrC)rYr:r`r rr?rrr rDzSampler.get_cellrErrr?rr`s   r`c@s4eZdZdZgdZ d ddZddZd d ZdS) MonitoredTimerzGA context manager to measure the walltime and increment a Counter cell.)r*tmonitored_section_name _counting_avoid_repetitive_countingNFcCs||_||_||_d|_dS)aVCreates a new MonitoredTimer. Args: cell: the cell associated with the time metric that will be inremented. monitored_section_name: name of action being monitored here. avoid_repetitive_counting: when set to True, if already in a monitored timer section with the same monitored_section_name, skip counting. TN)r*rdrfre)rr*rdZavoid_repetitive_countingrrrrs  zMonitoredTimer.__init__cCs@|jr|jr|jtvrd|_|St|_|jrt|j|SNF)rfrdMonitoredTimerSectionsretimercappendr4rrr __enter__s  zMonitoredTimer.__enter__cCsL~~~|jr"t|jd}|jt||jr$t|jdSdSdS)Ni@B) rerircr*r/intrdrhremove)rZexception_typeZexception_value tracebackZ micro_secondsrrr__exit__szMonitoredTimer.__exit__rg)rr r!r"r#rrkrorrrrrbs  rbcsfdd}|S)zA function decorator for adding MonitoredTimer support. Args: cell: the cell associated with the time metric that will be inremented. Returns: A decorator that measure the function runtime and increment the specified counter cell. cstfdd}|S)Ncs:t|i|WdS1swYdSr)rb)rkwargs)r*funcrrwrappers  $z:monitored_timer..actual_decorator..wrapper) functoolswraps)rqrrr*)rqractual_decoratorsz)monitored_timer..actual_decoratorr)r*rvrrurmonitored_timer s  rw)[r" collectionsrsriZtensorflow.core.frameworkrZtensorflow.pythonrZtensorflow.python.clientrZtensorflow.python.frameworkrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportr namedtupleZ _MetricMethodZTFE_MonitoringNewCounter0ZTFE_MonitoringDeleteCounter0ZTFE_MonitoringGetCellCounter0ZTFE_MonitoringNewCounter1ZTFE_MonitoringDeleteCounter1ZTFE_MonitoringGetCellCounter1ZTFE_MonitoringNewCounter2ZTFE_MonitoringDeleteCounter2ZTFE_MonitoringGetCellCounter2r;ZTFE_MonitoringNewIntGauge0ZTFE_MonitoringDeleteIntGauge0ZTFE_MonitoringGetCellIntGauge0ZTFE_MonitoringNewIntGauge1ZTFE_MonitoringDeleteIntGauge1ZTFE_MonitoringGetCellIntGauge1ZTFE_MonitoringNewIntGauge2ZTFE_MonitoringDeleteIntGauge2ZTFE_MonitoringGetCellIntGauge2rLZTFE_MonitoringNewStringGauge0Z TFE_MonitoringDeleteStringGauge0Z!TFE_MonitoringGetCellStringGauge0ZTFE_MonitoringNewStringGauge1Z TFE_MonitoringDeleteStringGauge1Z!TFE_MonitoringGetCellStringGauge1ZTFE_MonitoringNewStringGauge2Z TFE_MonitoringDeleteStringGauge2Z!TFE_MonitoringGetCellStringGauge2ZTFE_MonitoringNewStringGauge3Z TFE_MonitoringDeleteStringGauge3Z!TFE_MonitoringGetCellStringGauge3ZTFE_MonitoringNewStringGauge4Z TFE_MonitoringDeleteStringGauge4Z!TFE_MonitoringGetCellStringGauge4rSZTFE_MonitoringNewBoolGauge0ZTFE_MonitoringDeleteBoolGauge0ZTFE_MonitoringGetCellBoolGauge0ZTFE_MonitoringNewBoolGauge1ZTFE_MonitoringDeleteBoolGauge1ZTFE_MonitoringGetCellBoolGauge1ZTFE_MonitoringNewBoolGauge2ZTFE_MonitoringDeleteBoolGauge2ZTFE_MonitoringGetCellBoolGauge2rXZTFE_MonitoringNewSampler0ZTFE_MonitoringDeleteSampler0ZTFE_MonitoringGetCellSampler0ZTFE_MonitoringNewSampler1ZTFE_MonitoringDeleteSampler1ZTFE_MonitoringGetCellSampler1ZTFE_MonitoringNewSampler2ZTFE_MonitoringDeleteSampler2ZTFE_MonitoringGetCellSampler2raobjectr r%r8rHrKrMrRrUrWrYr\r^r`rhrbrwrrrrs       + # 3