BVhdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z ddl mZdd lmZe j"Zd ed d fd Zd ed efdZd ed efdZd ed efdZde j.d ej0fdZy )zUtilities for SavedModel fingerprinting. This module contains utility classes and functions for working with the SavedModel fingerprint. )logging)flags)fingerprint_pb2)file_io)fingerprinting) constants)metrics)compat export_dirreturnNcztjjjrt j t j|t jtj}tjd| tj|}t j ||t#j$| t#j&|t)|yy#t$r}t|dd}~wwxYw#t"j*$rtjdYywxYw)a!Write fingerprint protobuf, if requested. Writes a `tf.saved_model.experimental.Fingerprint` object to a `fingerprint.pb` file in the `export_dir` using the `saved_model.pb` file contained in `export_dir`. Args: export_dir: The directory in which to write the fingerprint. zWriting fingerprint to %sN fingerprint)path singleprintzIpath_and_singleprint metric could not be set. Model saving will continue.)rconfigsaved_model_fingerprintingvaluerjoinr as_strrFINGERPRINT_FILENAMErinfofingerprinting_pywrapCreateFingerprintDefFingerprintException ValueErroratomic_write_string_to_filer SetWriteFingerprintSetWritePathAndSingleprint"singleprint_from_fingerprint_protoMetricException)r fingerprint_pathfingerprint_serializedes b/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/saved_model/fingerprinting_utils.pywrite_fingerprintr&#s \\^..446|| j! i4457 LL,.>?$4II   ''(8(>@ ,BC2 ((8DF!7 $ qMt#$  " "2 ll122s*C2 D2 D ; DD (D:9D:ch tj|S#t$r}t|dd}~wwxYw)aCReturns the singleprint of `saved_model.pb` in `export_dir`. Args: export_dir: The directory that contains `saved_model.pb`. Returns: A string containing the singleprint of `saved_model.pb` in `export_dir`. Raises: ValueError: If a valid singleprint cannot be constructed from `saved_model.pb`. N)rSingleprintFromSMrrr r$s r%"singleprint_from_saved_model_protor*E4" 2 2: >> " Q-T!" 1 ,1ch tj|S#t$r}t|dd}~wwxYw)aBReturns the singleprint of `fingerprint.pb` in `export_dir`. Args: export_dir: The directory that contains `fingerprint.pb`. Returns: A string containing the singleprint of `fingerprint.pb` in `export_dir`. Raises: ValueError: If a valid singleprint cannot be constructed from `fingerprint.pb`. N)rSingleprintFromFPrrr)s r%r r Xr+r,c t|S#t$rYnwxYw t|t|S#t$rYnwxYw t|S#t$r}t|dd}~wwxYw)aReturns the singleprint of the SavedModel in `export_dir`. First tries to construct the singleprint from `fingerprint.pb`, then from `saved_model.pb`. Attempts to write the `fingerprint.pb` if not found, but doesn't return an error if it isn't writeable. Args: export_dir: The directory that contains the SavedModel. Returns: A string containing the singleprint of the SavedModel in `export_dir`. Raises: ValueError: If a valid singleprint cannot be constructed from the SavedModel. N)r rr&r*r)s r%singleprint_from_saved_modelr0ksz$ -j 99    j! -j 99   " -j 99 " Q-T!"s0 3 ?? A A( A##A(rctj|j|j|j|j |j S)N)saved_model_checksumgraph_def_program_hashsignature_def_hashsaved_object_graph_hashcheckpoint_hash)rFingerprintDefr2r3r4r5r6rs r%to_protor8sB  ' '&;;(??$77)AA!11  33)__doc__abslrtensorflow.core.configrtensorflow.core.protobufrtensorflow.python.lib.iortensorflow.python.saved_modelr0tensorflow.python.saved_model.pywrap_saved_modelrrr tensorflow.python.utilr rstrr&r*r r0 Fingerprintr7r8r9r%rEs (4,8FdD),AA2#2$2D"3"3"&"3"3"&""S""S""J3++30?0N0N3r9