BVh'A JdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZej d d Zeej$ej&ej( eej*ej,ej. eej0ej2ej4 gZeej8ej:ej< eej>ej@ejB eejDejFejH gZ%eejLejNejP eejRejTejV eejXejZej\ eej^ej`ejb eejdejfejh gZ5eejlejnejp eejrejtejv eejxejzej| gZ?eejejej eejejej eejejej gZIGd d eJZKGddeJZLGddeKZMGddeJZNGddeKZOGddeJZPGddeKZQGddeJZRedgGddeKZSGd d!eJZTGd"d#eJZUGd$d%eUZVGd&d'eKZWgZXGd(d)eJZYd*ZZy)+zTensorFlow monitoring APIs.N) summary_pb2) pywrap_tfe)pywrap_tf_session) c_api_util)compat) tf_export MetricMethodzcreate delete get_cell)createdeleteget_cellc*eZdZdZgdZdZdZdZy)MetriczThe base class of metric.)_metric _metric_name_metric_methods _label_lengthc$||_||_||_|t|jk\r8t dj |jt|j|j|jj ||_y)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_lengthargss R/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/eager/monitoring.py__init__zMetric.__init__rs$D)D%Ds4//00 AHH   S!5!568 99C4''(:(:;BBDIDLc |j|jj}|j}| ||yy#t$rYywxYwN)rrr rAttributeError)rdeletermetrics r__del__zMetric.__del__sT $$T%7%78??g||f fo    s/= A A ct||jk7r/tdj|j|j|j |jj |jg|S)Retrieves the cell.zThe {} expects taking {} labels)rrrrrrr r)rlabelss rr zMetric.get_cellsu 6{d((( 8??   T//1 22 <4   2 2 3 < <   rN)__name__ __module__ __qualname____doc__ __slots__rr$r rrrrms!M)J&rrc(eZdZdZdgZdZdZdZy) CounterCellz+CounterCell stores each value of a Counter._cellc||_y)z`Creates a new CounterCell. Args: cell: A c pointer of TFE_MonitoringCounterCell. Nr0rcells rrzCounterCell.__init__ DJrcDtj|j|y)zQAtomically increments the value. Args: value: non-negative value. N)r$TFE_MonitoringCounterCellIncrementByr0rvalues r increase_byzCounterCell.increase_bys 33DJJFrc@tj|jSRetrieves the current value.)rTFE_MonitoringCounterCellValuer0rs rr9zCounterCell.values  4 4TZZ @@rN)r(r)r*r+r,rr:r9r-rrr/r/s3i)GArr/c0eZdZdZgZfdZfdZxZS)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. cPtt| dtt |||g|y)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. rAN)superrAr_counter_methodsrrname descriptionr' __class__s rrzCounter.__init__s, '4!)-=s6{"& >6<>rc6ttt||Sr&)r/rCrAr rr'rHs rr zCounter.get_cell uWd4f= >>rr(r)r*r+r,rr __classcell__rHs@rrArAs) >??rrAc(eZdZdZdgZdZdZdZy) IntGaugeCellz/A single integer value stored in an `IntGauge`.r0c||_y)zbCreates a new IntGaugeCell. Args: cell: A c pointer of TFE_MonitoringIntGaugeCell. Nr2r3s rrzIntGaugeCell.__init__r5rcDtj|j|y)zEAtomically set the value. Args: value: integer value. N)rTFE_MonitoringIntGaugeCellSetr0r8s rsetzIntGaugeCell.sets ,,TZZ?rc@tj|jSr<)rTFE_MonitoringIntGaugeCellValuer0r?s rr9zIntGaugeCell.values  5 5djj AArNr(r)r*r+r,rrUr9r-rrrQrQs7i)@BrrQc0eZdZdZgZfdZfdZxZS)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. cPtt| dtt |||g|y)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. rZN)rCrZr_int_gauge_methodsrrEs rrzIntGauge.__init__s, (D":/A3v;#'?7=?rc6ttt||SrJ)rQrCrZr rKs rr zIntGauge.get_cells h6? @@rrMrOs@rrZrZs ) ?AArrZc(eZdZdZdgZdZdZdZy)StringGaugeCellz1A single string value stored in an `StringGauge`.r0c||_y)zhCreates a new StringGaugeCell. Args: cell: A c pointer of TFE_MonitoringStringGaugeCell. Nr2r3s rrzStringGaugeCell.__init__r5rcDtj|j|y)zDAtomically set the value. Args: value: string value. N)r TFE_MonitoringStringGaugeCellSetr0r8s rrUzStringGaugeCell.set s // EBrctj5}tj|j|t j |jd}ddd|S#1swYSxYw)r=zutf-8N)r tf_bufferr"TFE_MonitoringStringGaugeCellValuer0r TF_GetBufferdecode)rbuffer_r9s rr9zStringGaugeCell.values`    F733DJJH,,W5<rc2tj5}tj|j|t j |}dddtj}|jtj|S#1swYCxYw)zRetrieves the current distribution of samples. Returns: A HistogramProto describing the distribution of samples. N) rrdrTFE_MonitoringSamplerCellValuer0rrfrHistogramProtoParseFromStringras_bytes)rrh proto_datahistogram_protos rr9zSamplerCell.valuesx    ;7// GD$11':j;"002O##FOOJ$?@  ;;s 6B  BN)r(r)r*r+r,rrr9r-rrr|r|ls3i)? rr|c"eZdZdZdgZdZdZy)Bucketsz&Bucketing strategies for the samplers.bucketsc||_y)z[Creates a new Buckets. Args: buckets: A c pointer of TFE_MonitoringBuckets. N)r)rrs rrzBuckets.__init__s DLrcBtj|jyr )rTFE_MonitoringDeleteBucketsrr?s rr$zBuckets.__del__s**4<<8rN)r(r)r*r+r,rr$r-rrrrs.k)9rrc&eZdZdZgZfdZxZS)ExponentialBucketszExponential bucketing strategy. Sets up buckets of the form: [-DBL_MAX, ..., scale * growth^i, scale * growth_factor^(i + 1), ..., DBL_MAX]. cVtt| tj|||y)z|Creates a new exponential Buckets. Args: scale: float growth_factor: float bucket_count: integer N)rCrrr#TFE_MonitoringNewExponentialBuckets)rscale growth_factor bucket_countrHs rrzExponentialBuckets.__init__s, d,66um7C EFr)r(r)r*r+r,rrNrOs@rrrs) F Frrc0eZdZdZgZfdZfdZxZS)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. cftt| dtt |||j |g|y)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. rN)rCrr_sampler_methodsrr)rrFrrGr'rHs rrzSampler.__init__s4 '4!)-=s6{"&OGMOrc6ttt||SrJ)r|rCrr rKs rr zSampler.get_cellrLrrMrOs@rrrs) O??rrc.eZdZdZgdZ ddZdZdZy)MonitoredTimerzGA context manager to measure the walltime and increment a Counter cell.)r4tmonitored_section_name _counting_avoid_repetitive_countingNc<||_||_||_d|_y)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)r4rrr)rr4ravoid_repetitive_countings rrzMonitoredTimer.__init__s"DI"8D&?D#DNrc|jr'|jr|jtvr d|_|St j|_|jrtj |j|SNF)rrMonitoredTimerSectionsrtimerappendr?s r __enter__zMonitoredTimer.__enter__s^ ''  ' '  ' '+A Adn k YY[DF ""##D$?$?@ Krc ~~~|jrutj|jz dz}|jj t ||j r tj|j yyy)Ni@B) rrrr4r:intrrremove)rexception_typeexception_value traceback micro_secondss r__exit__zMonitoredTimer.__exit__sg ~~yy{TVV+w6m iiC ./ $ $%%d&A&AB %rr)r(r)r*r+r,rrrr-rrrrs#O)JO Crrcfd}|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. cFtjfd}|S)NcVt5|i|cdddS#1swYyxYwr )r)rkwargsr4funcs rwrapperz:monitored_timer..actual_decorator..wrappers- $ %T$V$%%%s() functoolswraps)rrr4s` ractual_decoratorz)monitored_timer..actual_decorators%__T%% Nrr-)r4rs` rmonitored_timerr 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_MonitoringGetCellCounter2rDTFE_MonitoringNewIntGauge0TFE_MonitoringDeleteIntGauge0TFE_MonitoringGetCellIntGauge0TFE_MonitoringNewIntGauge1TFE_MonitoringDeleteIntGauge1TFE_MonitoringGetCellIntGauge1TFE_MonitoringNewIntGauge2TFE_MonitoringDeleteIntGauge2TFE_MonitoringGetCellIntGauge2r\TFE_MonitoringNewStringGauge0 TFE_MonitoringDeleteStringGauge0!TFE_MonitoringGetCellStringGauge0TFE_MonitoringNewStringGauge1 TFE_MonitoringDeleteStringGauge1!TFE_MonitoringGetCellStringGauge1TFE_MonitoringNewStringGauge2 TFE_MonitoringDeleteStringGauge2!TFE_MonitoringGetCellStringGauge2TFE_MonitoringNewStringGauge3 TFE_MonitoringDeleteStringGauge3!TFE_MonitoringGetCellStringGauge3TFE_MonitoringNewStringGauge4 TFE_MonitoringDeleteStringGauge4!TFE_MonitoringGetCellStringGauge4rlTFE_MonitoringNewBoolGauge0TFE_MonitoringDeleteBoolGauge0TFE_MonitoringGetCellBoolGauge0TFE_MonitoringNewBoolGauge1TFE_MonitoringDeleteBoolGauge1TFE_MonitoringGetCellBoolGauge1TFE_MonitoringNewBoolGauge2TFE_MonitoringDeleteBoolGauge2TFE_MonitoringGetCellBoolGauge2ryTFE_MonitoringNewSampler0TFE_MonitoringDeleteSampler0TFE_MonitoringGetCellSampler0TFE_MonitoringNewSampler1TFE_MonitoringDeleteSampler1TFE_MonitoringGetCellSampler1TFE_MonitoringNewSampler2TFE_MonitoringDeleteSampler2TFE_MonitoringGetCellSampler2robjectrr/rArQrZr_rjrorwr|rrrrrrr-rrrs" 1(62)6& &&~7OP 336699;336699;336699; 4477::<4477::<4477::< 77::==?77::==?77::==?77::==?77::==?#.5588;;=5588;;=5588;;= 336699;336699;336699;  (V(VA&A4?f?4B6B4AvA4f:G&G4CFC4 .26CC7C4 & F9f9"FF.?f?<0CV0Cfr