o 2h'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_export 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)rrrlen ValueErrorformatr r )self metric_namemetric_methods label_lengthargsr]/var/www/html/chatgem/venv/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)rdeletermetricrrr__del__s   zMetric.__del__cGs>t||jkrtd|j|j|j|jj|jg|RS)Retrieves the cell.zThe {} expects taking {} labels)rrrrrrr r rlabelsrrrr s zMetric.get_cellN)__name__ __module__ __qualname____doc__ __slots__rr r 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)r$TFE_MonitoringCounterCellIncrementByr+rvaluerrr increase_byzCounterCell.increase_bycC t|jSRetrieves the current value.)rTFE_MonitoringCounterCellValuer+rrrrr4 zCounterCell.valueN)r$r%r&r'r(rr5r4rrrrr*  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. r?N)superr?r_counter_methodsrrname descriptionr# __class__rrr  zCounter.__init__cttt|j|Sr!)r*rAr?r r"rFrrr zCounter.get_cellr$r%r&r'r(rr __classcell__rrrFrr?   r?c@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.rrrrr0zIntGaugeCell.__init__cCr1)zEAtomically set the value. Args: value: integer value. N)rTFE_MonitoringIntGaugeCellSetr+r3rrrsetr6zIntGaugeCell.setcCr7r8)rTFE_MonitoringIntGaugeCellValuer+r;rrrr4r<zIntGaugeCell.valueNr$r%r&r'r(rrQr4rrrrrOr=rOcr>)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. cr@)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. rTN)rArTr_int_gauge_methodsrrCrFrrrrHzIntGauge.__init__crIrJ)rOrArTr r"rFrrr rKzIntGauge.get_cellrLrrrFrrTrNrTc@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.rrrrr0zStringGaugeCell.__init__cCr1)zDAtomically set the value. Args: value: string value. N)r TFE_MonitoringStringGaugeCellSetr+r3rrrrQ r6zStringGaugeCell.setcCsLt}t|j|t|d}Wd|S1swY|S)r9zutf-8N)r tf_bufferr"TFE_MonitoringStringGaugeCellValuer+r TF_GetBufferdecode)rbuffer_r4rrrr4s  zStringGaugeCell.valueNrSrrrrrVr=rVcr>) 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. cr@)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. r]N)rAr]r_string_gauge_methodsrrCrFrrr'   zStringGauge.__init__crIrJ)rVrAr]r r"rFrrr 2rKzStringGauge.get_cellrLrrrFrr]rNr]c@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<r0zBoolGaugeCell.__init__cCr1)zBAtomically set the value. Args: value: bool value. N)rTFE_MonitoringBoolGaugeCellSetr+r3rrrrQDr6zBoolGaugeCell.setcCr7r8)r TFE_MonitoringBoolGaugeCellValuer+r;rrrr4Lr<zBoolGaugeCell.valueNrSrrrrr`7r=r`z!__internal__.monitoring.BoolGauge)v1cr>) 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. cr@)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. rdN)rArdr_bool_gauge_methodsrrCrFrrr\r_zBoolGauge.__init__crIrJ)r`rArdr r"rFrrr grKzBoolGauge.get_cellrLrrrFrrdQs   rdc@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.rrrrqr0zSamplerCell.__init__cCr1)zBAtomically add a sample. Args: value: float value. N)rTFE_MonitoringSamplerCellAddr+r3rrraddyr6zSamplerCell.addcCs\t}t|j|t|}Wdn1swYt}| t ||S)zRetrieves the current distribution of samples. Returns: A HistogramProto describing the distribution of samples. N) rrXrTFE_MonitoringSamplerCellValuer+rrZrHistogramProtoParseFromStringras_bytes)rr\ proto_datahistogram_protorrrr4s  zSamplerCell.valueN)r$r%r&r'r(rrhr4rrrrrflr=rfc@s&eZdZdZdgZddZddZdS)Bucketsz&Bucketing strategies for the samplers.bucketscCr,)z[Creates a new Buckets. Args: buckets: A c pointer of TFE_MonitoringBuckets. N)rp)rrprrrrr0zBuckets.__init__cCst|jdSr)rTFE_MonitoringDeleteBucketsrpr;rrrr szBuckets.__del__N)r$r%r&r'r(rr rrrrros  rocs$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)rArrrr#TFE_MonitoringNewExponentialBuckets)rscale growth_factor bucket_countrFrrrs zExponentialBuckets.__init__)r$r%r&r'r(rrMrrrFrrrsrrcr>)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. rwN)rArwr_sampler_methodsrrp)rrDrprEr#rFrrrs   zSampler.__init__crIrJ)rfrArwr r"rFrrr rKzSampler.get_cellrLrrrFrrws   rwc@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/r{r}r|)rr/r{avoid_repetitive_countingrrrrs  zMonitoredTimer.__init__cCs@|jr|jr|jtvrd|_|St|_|jrt|j|SNF)r}r{MonitoredTimerSectionsr|timerzappendr;rrr __enter__s   zMonitoredTimer.__enter__cCsL~~~|jr"t|jd}|jt||jr$t|jdSdSdS)Ni@B) r|rrzr/r5intr{rremove)rexception_typeexception_value traceback micro_secondsrrr__exit__szMonitoredTimer.__exit__r)r$r%r&r'r(rrrrrrrrys  rycsfdd}|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)ry)rkwargs)r/funcrrwrappers  $z:monitored_timer..actual_decorator..wrapper) functoolswraps)rrr/)rractual_decoratorsz)monitored_timer..actual_decoratorr)r/rrrrmonitored_timer s  r)[r' collectionsrrtensorflow.core.frameworkrtensorflow.pythonrtensorflow.python.clientrtensorflow.python.frameworkrtensorflow.python.utilr tensorflow.python.util.tf_exportr namedtuple _MetricMethodTFE_MonitoringNewCounter0TFE_MonitoringDeleteCounter0TFE_MonitoringGetCellCounter0TFE_MonitoringNewCounter1TFE_MonitoringDeleteCounter1TFE_MonitoringGetCellCounter1TFE_MonitoringNewCounter2TFE_MonitoringDeleteCounter2TFE_MonitoringGetCellCounter2rBTFE_MonitoringNewIntGauge0TFE_MonitoringDeleteIntGauge0TFE_MonitoringGetCellIntGauge0TFE_MonitoringNewIntGauge1TFE_MonitoringDeleteIntGauge1TFE_MonitoringGetCellIntGauge1TFE_MonitoringNewIntGauge2TFE_MonitoringDeleteIntGauge2TFE_MonitoringGetCellIntGauge2rUTFE_MonitoringNewStringGauge0 TFE_MonitoringDeleteStringGauge0!TFE_MonitoringGetCellStringGauge0TFE_MonitoringNewStringGauge1 TFE_MonitoringDeleteStringGauge1!TFE_MonitoringGetCellStringGauge1TFE_MonitoringNewStringGauge2 TFE_MonitoringDeleteStringGauge2!TFE_MonitoringGetCellStringGauge2TFE_MonitoringNewStringGauge3 TFE_MonitoringDeleteStringGauge3!TFE_MonitoringGetCellStringGauge3TFE_MonitoringNewStringGauge4 TFE_MonitoringDeleteStringGauge4!TFE_MonitoringGetCellStringGauge4r^TFE_MonitoringNewBoolGauge0TFE_MonitoringDeleteBoolGauge0TFE_MonitoringGetCellBoolGauge0TFE_MonitoringNewBoolGauge1TFE_MonitoringDeleteBoolGauge1TFE_MonitoringGetCellBoolGauge1TFE_MonitoringNewBoolGauge2TFE_MonitoringDeleteBoolGauge2TFE_MonitoringGetCellBoolGauge2reTFE_MonitoringNewSampler0TFE_MonitoringDeleteSampler0TFE_MonitoringGetCellSampler0TFE_MonitoringNewSampler1TFE_MonitoringDeleteSampler1TFE_MonitoringGetCellSampler1TFE_MonitoringNewSampler2TFE_MonitoringDeleteSampler2TFE_MonitoringGetCellSampler2rxobjectr r*r?rOrTrVr]r`rdrfrorrrwrryrrrrrs       + # 3