AVhdZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Zddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddl#m$Z%ddl&m'Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl-m/Z/ddl0m1Z1dZ2GddejfZ4e4Z5Gd d!Z6d"Z7e1d#g$d%Z8d&Z9d'Z:e1d(g$e/jvd)Zd,Z?d-Z@e1d.g$d/ZAe1d0g$d1ZBe1d2g$Gd3d4ej5ZDGd6d7eDZEGd8d9eFeEeFe*jZHGd:d;eEe*jeH5ZIGd<d=eDZJGd>d?eDZKe1d@g$ dsdAZLe1dBg$ dtdCZM dudDZNe1dEg$dFZOdGZPdHZQe1dIg$dJZRdKZSejdLZUe1dMg$e/jvdvdNZVe1dOg$dwdPZWe1dQg$dsdRZXdrdSZYdwdTZZdsdUZ[dsdVZ\dxdWZ]dsdXZ^dsdYZ_e1dZg$d[Z`drd\Zae1d]g$dsd^Zbdsd_Zcdrd`Zde.jddabdcZfddZgdeZhdfZidrdgZjdrdhZkejdidjZmejZodape1dkg$dydlZqe1dmg$dsdnZre1dog$dpZsdqZty)zzOperations to emit summaries.N) graph_pb2) summary_pb2) config_pb2)api)layout)context) constant_op)dtypes)ops) smart_cond)tensor) tensor_util) array_ops)control_flow_ops)gen_resource_variable_ops)gen_summary_ops)math_ops)resource_variable_ops)summary_op_util) tf_logging) profiler_v2)resource) training_util) deprecation) tf_contextlib) tf_export_SUMMARY_WRITER_V2ceZdZfdZxZS) _SummaryStatecbtt| d|_d|_d|_d|_y)NT)superr__init__ is_recording"is_recording_distribution_strategywriterstep)self __class__s T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/summary_ops_v2.pyr"z_SummaryState.__init__<s. -')D.2D+DKDI)__name__ __module__ __qualname__r" __classcell__r(s@r)rr:s r*rc$eZdZdZddZdZdZy)_SummaryContextManagerz8Context manager to implement SummaryWriter.as_default().Nc<||_||_d|_d|_yN)_writer_step _old_writer _old_step)r'r%r&s r)r"z_SummaryContextManager.__init__NsDLDJDDNr*ctj|_|jt_|j*tj |_|jt_|jSr3)_summary_stater%r6r4r5r&r7r's r) __enter__z _SummaryContextManager.__enter__TsI%,,D LLN zz%**dn JJn <<r*ctjj|jt_|j|j t_yNF)r9r%flushr6r5r7r&)r'excs r)__exit__z_SummaryContextManager.__exit__\s=! ,,N zz NNn r*r3)r+r,r-__doc__r"r;r@r*r)r1r1Hs@  r*r1ctjtjdSt tj sz3_should_record_summaries_internal..}sXa[acar*) r9r%r constantrEr#rconstant_valuer$r logical_and) default_state static_condresolvecond_distributedconds r)!_should_record_summaries_internalrPfs"    && .-- .,,^-H-HIK{  ! !% (( /'^NNO ,, -$ \ D   . 55r*zsummary.should_record_summaries)v1ctdS)aReturns boolean Tensor which is True if summaries will be recorded. If no default summary writer is currently registered, this always returns False. Otherwise, this reflects the recording condition has been set via `tf.summary.record_if()` (except that it may return False for some replicas when using `tf.distribute.Strategy`). If no recording condition is active, it defaults to True. TrKrPrBr*r)should_record_summariesrUs + >>r*ctdS)zEReturns boolean Tensor which is true if summaries should be recorded.FrSrTrBr*r)'_legacy_contrib_should_record_summariesrWs * ??r*cJtjduxrtjS)zFReturns non-Tensor boolean indicating if summaries are being recorded.Nr9r#rBr*r)is_recording_summariesrZs  $ $D 0 P^5P5PPr*zsummary.record_ifc#Ktj} |t_d|t_y#|t_wxYww)aSets summary recording on or off per the provided boolean value. The provided value can be a python boolean, a scalar boolean Tensor, or or a callable providing such a value; if a callable is passed it will be invoked on-demand to determine whether summary writing will occur. Note that when calling record_if() in an eager mode context, if you intend to provide a varying condition like `step % 100 == 0`, you must wrap this in a callable to avoid immediate eager evaluation of the condition. In particular, using a callable is the only way to have your condition evaluated as part of the traced body of an @tf.function that is invoked from within the `record_if()` context. Args: condition: can be True, False, a bool Tensor, or a callable providing such. Yields: Returns a context manager that sets this value on enter and restores the previous value on exit. NrY) conditionolds r) record_ifr^s3, ###&"+N "%N#Ns?/ ? <?c&tjduS)zEReturns a boolean indicating whether a default summary writer exists.N)r9r%rBr*r)has_default_writerr`s   d **r*ctjtjd5fd}t j s|}dddt |S#1swYt SxYw)zHSets the should_record_summaries Tensor to true if global_step % n == 0.Ncpu:0c6tjzdS)Nr)requal) global_stepnsr)rGz7record_summaries_every_n_global_steps..sX^^K!OQ7r*)rget_or_create_global_stepr devicerexecuting_eagerlyr^)rfreshoulds`` r)%record_summaries_every_n_global_stepsrkse99;K zz' 7F  $ $ &xf 6   6 s "A##A6ctdS)z7Sets the should_record_summaries Tensor to always true.Tr^rBr*r)always_record_summariesrns 4r*ctdS)z8Sets the should_record_summaries Tensor to always false.FrmrBr*r)never_record_summariesrps 5 r*zsummary.experimental.get_stepc"tjS)zReturns the default summary step for the current thread. Returns: The step set by `tf.summary.experimental.set_step()` if one has been set, otherwise None. r9r&rBr*r)get_steprss   r*zsummary.experimental.set_stepc|t_y)aSets the default summary step for the current thread. For convenience, this function sets a default value for the `step` parameter used in summary-writing functions elsewhere in the API so that it need not be explicitly passed in every such invocation. The value can be a constant or a variable, and can be retrieved via `tf.summary.experimental.get_step()`. Note: when using this with @tf.functions, the step value will be captured at the time the function is traced, so changes to the step outside the function will not be reflected inside the function unless using a `tf.Variable` step. Args: step: An `int64`-castable default step value, or None to unset. Nrrr&s r)set_steprvs .r*zsummary.SummaryWriterc2eZdZdZddZddZdZdZdZy) SummaryWriterz8Interface representing a stateful summary writer object.NcB|j|jy)aEnables this summary writer for the current thread. For convenience, if `step` is not None, this function also sets a default value for the `step` parameter used in summary-writing functions elsewhere in the API so that it need not be explicitly passed in every such invocation. The value can be a constant or a variable. Note: when setting `step` in a @tf.function, the step value will be captured at the time the function is traced, so changes to the step outside the function will not be reflected inside the function unless using a `tf.Variable` step. Args: step: An `int64`-castable default step value, or `None`. When not `None`, the current step is modified to the given value. When `None`, the current step is not modified. N) as_defaultr;r'r&s r)set_as_defaultzSummaryWriter.set_as_defaults$ OOD##%r*ct||S)a3Returns a context manager that enables summary writing. For convenience, if `step` is not None, this function also sets a default value for the `step` parameter used in summary-writing functions elsewhere in the API so that it need not be explicitly passed in every such invocation. The value can be a constant or a variable. Note: when setting `step` in a @tf.function, the step value will be captured at the time the function is traced, so changes to the step outside the function will not be reflected inside the function unless using a `tf.Variable` step. For example, `step` can be used as: ```python with writer_a.as_default(step=10): tf.summary.scalar(tag, value) # Logged to writer_a with step 10 with writer_b.as_default(step=20): tf.summary.scalar(tag, value) # Logged to writer_b with step 20 tf.summary.scalar(tag, value) # Logged to writer_a with step 10 ``` Args: step: An `int64`-castable default step value, or `None`. When not `None`, the current step is captured, replaced by a given one, and the original one is restored when the context manager exits. When `None`, the current step is not modified (and not restored when the context manager exits). Returns: The context manager. )r1r{s r)rzzSummaryWriter.as_default s@ "$ --r*ct)zInitializes the summary writer.NotImplementedErrorr:s r)initzSummaryWriter.init.  r*ct)zFlushes any buffered data.rr:s r)r>zSummaryWriter.flush2rr*ct)z&Flushes and closes the summary writer.rr:s r)closezSummaryWriter.close6rr*r3) r+r,r-rAr|rzrr>rrBr*r)rxrxs@&( .D   r*rx) metaclasscPeZdZdZd dZdZd fd Zd fd ZdZdZ dZ xZ S) _ResourceSummaryWriterzHImplementation of SummaryWriter using a SummaryWriterInterface resource.c|Ptj|j5||_||j|_dddn#||_||j|_d|_t jr|j||_ ytjt|j||_ y#1swYhxYwr=) dtensor_api default_mesh host_mesh _resource_init_op_closedrri_set_up_resource_deleterr add_to_collection$_SUMMARY_WRITER_INIT_COLLECTION_NAME_meshr' create_fn init_op_fnmeshs r)r"z_ResourceSummaryWriter.__init__>s   # #DNN$4 53""4>>2 33!{dn 0dmDL  " ##%DJ @$--PDJ33s $CCcPtj|jd|_yNrb)handle handle_device)rEagerResourceDeleterr_resource_deleterr:s r)rz/_ResourceSummaryWriter._set_up_resource_deleterQs2GG~~W6Dr*ctjr|jrtd|dt||y)z#See `SummaryWriter.set_as_default`.SummaryWriter  is already closedN)rrir RuntimeErrorr!r|r'r&r(s r)r|z%_ResourceSummaryWriter.set_as_defaultUs7  "t|| >$1CD EE G4 r*ctjr|jrtd|dt||S)zSee `SummaryWriter.as_default`.rr)rrirrr!rzrs r)rzz!_ResourceSummaryWriter.as_default[s:  "t|| >$1CD EE 7 d ##r*cxtjr|jrtd|d|jS)See `SummaryWriter.init`.rr)rrirrrr:s r)rz_ResourceSummaryWriter.initas3  "t|| >$1CD EE ==r*ctjr |jrytjd5t j |jcdddS#1swYyxYw)See `SummaryWriter.flush`.Nrb)rrirr rhrflush_summary_writerrr:s r)r>z_ResourceSummaryWriter.flushgsN  "t|| G B  1 1$.. ABBBs A  A)ctjr |jry tj|j g5tj d5tj|jcdddcdddtjrd|_SS#1swYnxYw dddn #1swYnxYwtjrd|_yy#tjrd|_wwxYw)See `SummaryWriter.close`.NrbT) rrirr control_dependenciesr>rhrclose_summary_writerrr:s r)rz_ResourceSummaryWriter.closens  "t||   # #TZZ\N 3F ZZ  F 55dnnE F FFF  " " $  % F F FFFF  " " $  % " " $  %sA$C*CB+< C C*+B4 0C8 C*C C**Dr3) r+r,r-rAr"rr|rzrr>rr.r/s@r)rr;s*P&6! $  B r*rc eZdZy)_MultiMetaclassN)r+r,r-rBr*r)rr{sr*rc0eZdZdZddZdZdZdZdZy) _TrackableResourceSummaryWriterzHA `_ResourceSummaryWriter` subclass that implements `TrackableResource`.Nctjjd|_|_t jfd||y)Nz/CPU:0)rhcjSr3)resource_handler:sr)rGz:_TrackableResourceSummaryWriter.__init__..s $..r*rrr)rTrackableResourcer" _create_fn _init_op_fnrrs` r)r"z(_TrackableResourceSummaryWriter.__init__sL ''X'>DO!D## .  $r*c"|jSr3)rr:s r)_create_resourcez0_TrackableResourceSummaryWriter._create_resources ?? r*c8|j|jSr3)rrr:s r) _initializez+_TrackableResourceSummaryWriter._initializes   D00 11r*cFtj|jdy)NT)ignore_lookup_error)rdestroy_resource_oprr:s r)_destroy_resourcez1_TrackableResourceSummaryWriter._destroy_resources11 $8r*cyr3rBr:s r)rz8_TrackableResourceSummaryWriter._set_up_resource_deleters r*r3) r+r,r-rAr"rrrrrBr*r)rrs!Q  28 r*rc(eZdZdZdZdZdZdZy)_LegacyResourceSummaryWriterzz"_LegacyResourceSummaryWriter.flushs9 G B  1 1$.. ABBBs ?Actj|jg5tjd5t j |j cdddcdddS#1swYnxYw dddy#1swYyxYw)rrbN)r rr>rhrrrr:s r)rz"_LegacyResourceSummaryWriter.closes ! !4::<. 1D ::g D33DNNCDDDDDDDDDDs"BA- B-A6 2BB N)r+r,r-rAr"rr>rrBr*r)rrsDK,B Dr*rcPeZdZdZddZej ddZdZdZ dZ y) _NoopSummaryWriterz=A summary writer that does nothing, for create_noop_writer().Ncyr3rBr{s r)r|z!_NoopSummaryWriter.set_as_defaultr*c#Kdywr3rBr{s r)rzz_NoopSummaryWriter.as_defaults  scyr3rBr:s r)rz_NoopSummaryWriter.initrr*cyr3rBr:s r)r>z_NoopSummaryWriter.flushrr*cyr3rBr:s r)rz_NoopSummaryWriter.closerr*r3) r+r,r-rAr|rcontextmanagerrzrr>rrBr*r)rrs3E       r*rzsummary.initializec|tjrytj t d|!t j }| td|jt|Nt|}tjtj}|jt|d||iyy)aInitializes summary writing for graph execution mode. This operation is a no-op when executing eagerly. This helper method provides a higher-level alternative to using `tf.contrib.summary.summary_writer_initializer_op` and `tf.contrib.summary.graph`. Most users will also want to call `tf.compat.v1.train.create_global_step` which can happen before or after this function is called. Args: graph: A `tf.Graph` or `tf.compat.v1.GraphDef` to output to the writer. This function will not write the default graph by default. When writing to an event log file, the associated step will be zero. session: So this method can call `tf.Session.run`. This defaults to `tf.compat.v1.get_default_session`. Raises: RuntimeError: If the current thread has no default `tf.contrib.summary.SummaryWriter`. ValueError: If session wasn't passed and no default session. Nz1No default tf.contrib.summary.SummaryWriter foundz=Argument `session must be passed if no default session existsr) feed_dict)rrir9r%rr get_default_session ValueErrorrunsummary_writer_initializer_op_serialize_graphr placeholderr stringgraph_v1)graphsessiondatarFs r) initializers6   " J KK _%%'G ( )) +++-.  E "Dfmm,A KKA1d)K4r*zsummary.create_file_writerc Z | tdtj}tjd5 tjd5tj 5t jrt|||||tj|tj}|tjd}|tjd}|tjd } fd }tjt j"||||} |r)t%|| | cdddcdddcdddSt'|| | cdddcdddcdddS#1swYnxYw dddn #1swYnxYwdddy#1swYyxYw) aCreates a summary file writer for the given log directory. Args: logdir: a string specifying the directory in which to write an event file. max_queue: the largest number of summaries to keep in a queue; will flush once the queue gets bigger than this. Defaults to 10. flush_millis: the largest interval between flushes. Defaults to 120,000. filename_suffix: optional suffix for the event file name. Defaults to `.v2`. name: a name for the op that creates the writer. experimental_trackable: a boolean that controls whether the returned writer will be a `TrackableResource`, which makes it compatible with SavedModel when used as a `tf.Module` property. experimental_mesh: a `tf.experimental.dtensor.Mesh` instance. When running with DTensor, the mesh (experimental_mesh.host_mesh()) will be used for bringing all the DTensor logging from accelerator to CPU mesh. Returns: A SummaryWriter object. Nz Argument `logdir` cannot be Nonecreate_file_writerrblogdir max_queue flush_millisfilename_suffixdtype .v2ctjrtj}ntj}t j |S)N) shared_namename)rrianonymous_namer name_from_scope_namersummary_writer)rrscopes r)rz(create_file_writer_v2..create_fn7sG  $ $ &..0+007+--#$0 0r*r)rr inside_function name_scoperh init_scoperri_check_create_file_writer_argsconvert_to_tensorr rr rH functoolspartialrcreate_summary_file_writerrr) rrrrrexperimental_trackableexperimental_meshrrrrs ` @r)create_file_writer_v2rs< ^ 7 88'')/ ~~d01) UCJJw>r*crtjr tdtjt S)zGraph-mode only. Returns the list of ops to create all summary writers. Returns: The initializer ops. Raises: RuntimeError: If in Eager mode. zQtf.contrib.summary.summary_writer_initializer_op is only supported in graph mode.)rrirr r rrBr*r)rrs6   # $$  @ AAr*z[^-_/.A-Za-z0-9]z"summary.experimental.summary_scopec#K|xs|}tj}|r|dz|zn|}tjd|xsd}tj|||d5}||fdddy#1swYyxYww)aExperimental context manager for use when defining a custom summary op. This behaves similarly to `tf.name_scope`, except that it returns a generated summary tag in addition to the scope name. The tag is structurally similar to the scope name - derived from the user-provided name, prefixed with enclosing name scopes if any - but we relax the constraint that it be uniquified, as well as the character set limitation (so the user-provided name can contain characters not legal for scope names; in the scope name these are removed). This makes the summary tag more predictable and consistent for the user. For example, to define a new summary op called `my_op`: ```python def my_op(name, my_value, step): with tf.summary.summary_scope(name, "MyOp", [my_value]) as (tag, scope): my_value = tf.convert_to_tensor(my_value) return tf.summary.write(tag, my_value, step=step) ``` Args: name: string name for the summary. default_name: Optional; if provided, used as default name of the summary. values: Optional; passed as `values` parameter to name_scope. Yields: A tuple `(tag, scope)` as described above. /NF) skip_on_eager)r get_name_scope_INVALID_SCOPE_CHARACTERSsubr)r default_namevalues current_scopetagrs r) summary_scopers~>  $$$&-&3 d"# # & &r4 0 8D$ ~~dL&F% u*sAA8A,# A8,A51A8z summary.writectj|d5tjt j dcdddS t |dnt|dr|jn|fd}tjt|td}tjs.tjtjj ||cdddS#1swYyxYw) aWrites a generic summary to the default SummaryWriter if one exists. This exists primarily to support the definition of type-specific summary ops like scalar() and image(), and is not intended for direct use unless defining a new type-specific summary op. Args: tag: string tag used to identify the summary (e.g. in TensorBoard), usually generated with `tf.summary.summary_scope` tensor: the Tensor holding the summary data to write or a callable that returns this Tensor. If a callable is passed, it will only be called when a default SummaryWriter exists and the recording condition specified by `record_if()` is met. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. metadata: Optional SummaryMetadata, as a proto or serialized bytes name: Optional string name for this op. Returns: True on success, or false if no summary was written because no default summary writer was available. Raises: ValueError: if a default writer exists, but no step was provided and `tf.summary.experimental.get_step()` is None. write_summaryNFr*SerializeToStringc  tdtjd5t r nt j }t j}t||}t|}tj|j||}tj|g5tjdcdddcdddS#1swYnxYw dddy#1swYyxYw)*Record the actual summary and return True.NpNo step set. Please specify one either through the `step` argument or through tf.summary.experimental.set_step()rbrT)rr rhrEridentityr9r% _maybe_convert_tensor_to_dtensorrrrrr rH) summary_tensorr% summary_value step_valuewrite_summary_oprserialized_metadatar&rr s r)recordzwrite..record s >? ? ::g ,%-f%59;M;M <  &&8P 5fdC *88          % %'7&8 9 ,%%d+ , ,%,,$ , , ,%,,,s$B C-/C C-C C--C6 summary_condr!)r rr9r%r rHrshasattrrr rUrrrirr r ) rr r&metadatarr)oprr(s ``` @@r)writer.s: ~~dO,-$  ! !% (-- | Zd . /$668$,,:   !68. JB  $ $ & CMM==rB [---s%C6 B!C66C?z!summary.experimental.write_raw_pbc8tj|d5tjt j dcdddSt  tdfd}tjd5tjt|td}tjs.tjtjj ||cdddcdddS#1swYnxYw dddy#1swYyxYw) a<Writes a summary using raw `tf.compat.v1.Summary` protocol buffers. Experimental: this exists to support the usage of V1-style manual summary writing (via the construction of a `tf.compat.v1.Summary` protocol buffer) with the V2 summary writing API. Args: tensor: the string Tensor holding one or more serialized `Summary` protobufs step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. name: Optional string name for this op. Returns: True on success, or false if no summary was written because no default summary writer was available. Raises: ValueError: if a default writer exists, but no step was provided and `tf.summary.experimental.get_step()` is None. write_raw_pbNFr cxtjd5tjtj j tj}tj|g5tjdcdddcdddS#1swYnxYw dddy#1swYyxYw)rrbr!TN) r rhrwrite_raw_proto_summaryr9r%rrr"rr rH)raw_summary_oprr&r s r)r)zwrite_raw_pb..recordOs ::g ,(@@  ! ! + +    v &   % %~&6 7 ,%%d+ , , ,,  , , , ,,,s$AB02B B0B# B00B9rbr*r!)r rr9r%r rHrsrrhr rUrrrirr r )r r&rr)r-rs`` @r)r0r0.s. ~~dN+u$  ! !% ( | Zd >? ? , G   ! #VXN Lb  & & ( cmm??D  -,-s*%D 5D>A)C:' D:D ?DDctjrdz fd}tjt j Stj d5tjt|td}tjs.tjtjj|ddd|S#1swYSxYw)a Helper function to write summaries. Args: name: name of the summary tensor: main tensor to form the summary function: function taking a tag and a scope which writes the summary family: optional, the summary's family Returns: The result of writing the summary. rcttj5tjg5\}}tj||g5t j dcdddcdddcdddS#1swYnxYw dddn #1swYnxYwdddy#1swYyxYw)N)rT)r rrrrr rH)rrfamilyfunctionrrr s r)r)z'summary_writer_function..recordss  #*_%B%B ffX&'**63  # #Xc5%9$: ;*##D)**************s@B.!BB' B0 B.B B B.B" B..B7Nrbrr!)r rr9r%rno_oprhr rWrrrirr r )rr r7r6r)r-rs```` @r)summary_writer_functionr9cs!!#*#J** "  ! ! ## zz'C   /168" NB  $ $ & CMM==rB C ) C )s #A(CCc4fd}t|||S)z$Writes a tensor summary if possible.c6tjd}n2tdr$tjj}n}t j t jjttj|||S)Nrrr!) r rHr+rrrr9r%r _choose_steprr")rrr(r,r&r s r)r7zgeneric..functions'004 . /'001K1K1MN$  ( (''T6"    r*r6r9)rr r,r6r&r7s `` ` r)genericr?s !vx GGr*c0fd}t|||S)aWrites a scalar summary if possible. Unlike `tf.contrib.summary.generic` this op may change the dtype depending on the writer, for both practical and efficiency concerns. Args: name: An arbitrary name for this summary. tensor: A `tf.Tensor` Must be one of the following types: `float32`, `float64`, `int32`, `int64`, `uint8`, `int16`, `int8`, `uint16`, `half`, `uint32`, `uint64`. family: Optional, the summary's family. step: The `int64` monotonic step variable, which defaults to `tf.compat.v1.train.get_global_step`. Returns: The created `tf.Operation` or a `tf.no_op` if summary writing has not been enabled for this context. ctjtjjt |t j|SNr!)rwrite_scalar_summaryr9r%rr<rr"rrr&r s r)r7zscalar..functionsA  / /''T 6"   r*r=r>rr r6r&r7s ` ` r)scalarrFs( !vx GGr*c0fd}t|||S)z'Writes a histogram summary if possible.ctjtjjt |t j|SrB)rwrite_histogram_summaryr9r%rr<rr"rDs r)r7zhistogram..functionsA  2 2''T 6"   r*r=r>rEs ` ` r) histogramrJs !vx GGr*c8fd}t|||S)z$Writes an image summary if possible.c 'tjgdtjn}t j t jjt|tj||S)N)rrrMrr!) r rHr uint8rwrite_image_summaryr9r%rr<rr")rr bad_color_ bad_color max_imagesr&r s r)r7zimage..functionsn&&&'7v||L,5  . .''T 6"  r*r=r>)rr rQrRr6r&r7s ``` ` r)imagerSs  !vx GGr*c8fd}t|||S)z$Writes an audio summary if possible.c tjtjjt |t j|S)N) sample_rate max_outputsr)rwrite_audio_summaryr9r%rr<rr")rrrWrVr&r s r)r7zaudio..functionsG  . .''T 6"  r*r=r>)rr rVrWr6r&r7s ``` ` r)audiorYs  !vx GGr*c|tjs?t|tjs%t d|dt |jdtj}|tjStjd5t|tjtj fr.tj"t%|t&j(}nt+j,|}t/j0|j2t5|||cdddS#1swYyxYw)aWrites a TensorFlow graph to the summary interface. The graph summary is, strictly speaking, not a summary. Conditions like `tf.summary.should_record_summaries` do not apply. Only a single graph can be associated with a particular run. If multiple graphs are written, then only the last one will be considered by TensorBoard. When not using eager execution mode, the user should consider passing the `graph` parameter to `tf.compat.v1.summary.initialize` instead of calling this function. Otherwise special care needs to be taken when using the graph to record the graph. Args: param: A `tf.Tensor` containing a serialized graph proto. When eager execution is enabled, this function will automatically coerce `tf.Graph`, `tf.compat.v1.GraphDef`, and string types. step: The global step variable. This doesn't have useful semantics for graph summaries, but is used anyway, due to the structure of event log files. This defaults to the global step. name: A name for the operation (optional). Returns: The created `tf.Operation` or a `tf.no_op` if summary writing has not been enabled for this context. Raises: TypeError: If `param` isn't already a `tf.Tensor` in graph mode. zigraph() needs a argument `param` to be tf.Tensor (e.g. tf.placeholder) in graph mode, but received param= of type .Nrbr!)rrir tensor_libTensor TypeErrortyper+r9r%rr8r rhGraphrGraphDefrrr rrr"rwrite_graph_summaryrr<)paramr&rr%r s r)rrs<  " " $Z Z  .  ye!5!5 6a 9   & ^  ! ! ## zz'A%#))Y%7%789$$%5e%$$ : & 7f E$YtJ/?/H/H.IL MM''      %%&&&sEB2EE cbtjtjj||S)aWrites a `tf.compat.v1.Event` binary proto. This can be used to import existing event logs into a new summary writer sink. Please note that this is lower level than the other summary functions and will ignore the `tf.summary.should_record_summaries` setting. Args: tensor: A `tf.Tensor` of type `string` containing a serialized `tf.compat.v1.Event` proto. name: A name for the operation (optional). Returns: The created `tf.Operation`. r!)r import_eventr9r%r)r rs r)rhrhbs*  % %%%vD ::r*z summary.flushc~|&tj}|tjSt |t r|j Std|)aForces summary writer to send any buffered data to storage. This operation blocks until that finishes. Args: writer: The `tf.summary.SummaryWriter` to flush. If None, the current default writer will be used instead; if there is no current writer, this returns `tf.no_op`. name: Ignored legacy argument for a name for the operation. Returns: The created `tf.Operation`. zInvalid argument to flush(): )r9r%rr8rrxr>rr%rs r)r>r>usO  ^  " "F ~  # # %% & <<>@AAr*c|t|tr t||Stjd5t j ||cdddS#1swYyxYw)aMLegacy version of flush() that accepts a raw resource tensor for `writer`. Do not use this function in any new code. Not supported and not part of the public TF APIs. Args: writer: The `tf.summary.SummaryWriter` to flush. If None, the current default writer will be used instead; if there is no current writer, this returns `tf.no_op`. For this legacy version only, also accepts a raw resource tensor pointing to the underlying C++ writer resource. name: Ignored legacy argument for a name for the operation. Returns: The created `tf.Operation`. Nrbr!)rrxr>r rhrrrjs r)legacy_raw_flushrlsV  ^z&-8   G E  1 1&t DEEEs AAcXtjj||sdSd|zS)z.Construct a logdir for an eval summary writer.evaleval_)ospathjoin) model_dirrs r)eval_dirrts% it HH4 HHr*z Renamed to create_file_writer().)date instructionscBtjdt|i|S)z3Please use `tf.contrib.summary.create_file_writer`.zQDeprecation Warning: create_summary_file_writer was renamed to create_file_writer)loggingwarningr)argskwargss r)rrs% //*+ T ,V ,,r*ct|tjr |jdj S|j S)NT) add_shapes)rr ra as_graph_defr)arbitrary_graphs r)rrs<+  ' '4 ' 8 J J LL  , , ..r*c|tjSt|tjs$t j |tjS|Sr3) rrgrr]r^r rr int64rus r)r<r<sB \  2 2 44 D*++ ,  v|| 44 +r*c |jD][\}}t|tjr!t j |s7|rt d|d|dt d|d|dy)a'Helper to check the validity of arguments to a create_file_writer() call. Args: inside_function: whether the create_file_writer() call is in a tf.function **kwargs: the arguments to check, as kwargs to give them names. Raises: ValueError: if the arguments are graph tensors. zInvalid graph Tensor argument '=z' to create_file_writer() inside an @tf.function. The create call will be lifted into the outer eager execution context, so it cannot consume graph tensors defined inside the function body.z+' to eagerly executed create_file_writer().N)itemsrr EagerTensorr is_tf_typer)rr{arg_nameargs r)rrs||~ .mh c3?? + 0F0Fs0K -hZq>F FG G -hZq>- -. . .r*ctj}d|j_d|j_t |d||g5\}}t jd5tj|jtj}dddt|||cdddS#1swY!xYw#1swYyxYw)aWrites entire RunMetadata summary. A RunMetadata can contain DeviceStats, partition graphs, and function graphs. Please refer to the proto for definition of each field. Args: name: A name for this summary. The summary tag used for TensorBoard will be this name prefixed by any active name scopes. data: A RunMetadata proto to write. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. Returns: True on success, or false if no summary was written because no default summary writer was available. Raises: ValueError: if a default writer exists, but no step was provided and `tf.summary.experimental.get_step()` is None. graph_run_metadata1graph_run_metadata_summaryrbrNrr r&r,)rSummaryMetadata plugin_data plugin_namecontentrr rhr rHrr rr.rrr&summary_metadatar_r s r) run_metadatars,!002.B*)-&T1D\# #&.sA G 9##D$:$:$<*0--9f9   !  # # #99 # #s$B?4B3B?3B< 8B??Cctj}d|j_d|j_t j |j|j}t|d||g5\}}tjd5tj|jtj }dddt#|||cdddS#1swY!xYw#1swYyxYw) aWrites graphs from a RunMetadata summary. Args: name: A name for this summary. The summary tag used for TensorBoard will be this name prefixed by any active name scopes. data: A RunMetadata proto to write. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. Returns: True on success, or false if no summary was written because no default summary writer was available. Raises: ValueError: if a default writer exists, but no step was provided and `tf.summary.experimental.get_step()` is None. graph_run_metadata_graphr)function_graphspartition_graphs graph_run_metadata_graph_summaryrbrNr)rrrrrr RunMetadatarrrr rhr rHrr rr.rs r)run_metadata_graphsrs&!002.H*)-&   **,, .$T7D\# #&.sA G 9##D$:$:$<*0--9f9   !  # # #99 # #s$1C* 4C>C*C' #C**C3 TraceContextrprofilerzsummary.trace_onc4tjrtjdyt j stjdyt 5trtjd dddy|r$|s"t jj|rO|tjdn7t jjtj|t||adddy#1swYyxYw)aStarts a trace to record computation graphs and profiling information. Must be invoked in eager mode. When enabled, TensorFlow runtime will collect information that can later be exported and consumed by TensorBoard. The trace is activated across the entire TensorFlow runtime and affects all threads of execution. To stop the trace and export the collected information, use `tf.summary.trace_export`. To stop the trace without exporting, use `tf.summary.trace_off`. Args: graph: If True, enables collection of executed graphs. It includes ones from tf.function invocation and ones from the legacy graph mode. The default is True. profiler: If True, enables the advanced profiler. Enabling profiler implicitly enables the graph collection. The profiler may incur a high memory overhead. The default is False. profiler_outdir: Output directory for profiler. It is required when profiler is enabled when trace was started. Otherwise, it is ignored. z)Cannot enable trace inside a tf.function.Nz Must enable trace in eager mode.zTrace already enabledzENo `profiler_outdir` passed to trace_on(). Profiler won't be enabled.r) r rrxwarnrri_current_trace_context_lock_current_trace_contextenable_graph_collectionenable_run_metadata _profilerstart _TraceContext)rrprofiler_outdirs r)trace_onr6s0  LL<=  " " $ LL34 #K ll*+ KK  X oo//1      --/(*J'KKKsDBDDzsummary.trace_exportctjrtjdyt j stjdyt 5t tdt\}}dddt jj}rst|||n t|||r+|rtjdtjty#1swYxYw)aStops and exports the active trace as a Summary and/or profile file. Stops the trace and exports all metadata collected during the trace to the default SummaryWriter, if one has been set. Args: name: A name for the summary to be written. step: Explicit `int64`-castable monotonic step value for this summary. If omitted, this defaults to `tf.summary.experimental.get_step()`, which must not be None. profiler_outdir: This arg is a no-op. Please set this in trace_on(). Raises: ValueError: if a default writer exists, but no step was provided and `tf.summary.experimental.get_step()` is None. z)Cannot export trace inside a tf.function.Nz.Can only export trace while executing eagerly.zrlrt deprecatedrrr<rrr namedtuplerLockrrrrrr#rBr*r)rs6 $  /1/8:+3.+2<3+2;1*71<?04.06 (<$IOO<6> ,4 ?5 ?@ Q  2&&'&8+    *r23 *r23$ "r*C ckkC +C L=]=@  $x'A'A"B %  % PD=D:  ( #$%  '5&'5T 'B/  I 0I Z"&$('+ -.` 'B/05 %  +,- ?. ? B 'BJJ':; /B7$8$N ?r"I#IX .26171h @H*H@ HH&H"/Ad ?r"A&#A&H:& ?r"B#B0E2I T%GI-I-/.0'#T(#V' &&~7LM ,inn. "%2K&2Kn !b)1*1h 2& ' *r*