o 2h@spdZddlmZddlmZddlmZddlm Z e dgdGdd d Z e d gdd e d e fd dZ dS)z{Methods for SavedModel fingerprinting. This module contains classes and functions for reading the SavedModel fingerprint. )Any)fingerprint_pb2)fingerprinting) tf_exportz$saved_model.experimental.Fingerprint)v1c @seZdZdZ      ddedededededef d d Zed ejd dfd dZ de d e fddZ d e fddZd e fddZd ejfddZdS) FingerprintayThe SavedModel fingerprint. Each attribute of this class is named after a field name in the FingerprintDef proto and contains the value of the respective field in the protobuf. Attributes: saved_model_checksum: A uint64 containing the `saved_model_checksum`. graph_def_program_hash: A uint64 containing `graph_def_program_hash`. signature_def_hash: A uint64 containing the `signature_def_hash`. saved_object_graph_hash: A uint64 containing the `saved_object_graph_hash`. checkpoint_hash: A uint64 containing the`checkpoint_hash`. version: An int32 containing the producer field of the VersionDef. Nsaved_model_checksumgraph_def_program_hashsignature_def_hashsaved_object_graph_hashcheckpoint_hashversioncCs(||_||_||_||_||_||_dS)aInitializes the instance based on values in the SavedModel fingerprint. Args: saved_model_checksum: Value of the`saved_model_checksum`. graph_def_program_hash: Value of the `graph_def_program_hash`. signature_def_hash: Value of the `signature_def_hash`. saved_object_graph_hash: Value of the `saved_object_graph_hash`. checkpoint_hash: Value of the `checkpoint_hash`. version: Value of the producer field of the VersionDef. N)rr r r r r )selfrr r r r r rg/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/saved_model/fingerprinting.py__init__-s  zFingerprint.__init__protoreturnc Cs`t|tr tj|}zt|j|j|j|j |j |j WSt y/}zt d|dd}~ww)z;Constructs Fingerprint object from protocol buffer message.z5Given proto could not be deserialized as fingerprint.N) isinstancebytesrFingerprintDef FromStringrrr r r r r AttributeError ValueError)clsrerrr from_protoGs*  zFingerprint.from_protoothercCsnt|ts t|tjr5z|j|jko)|j|jko)|j|jko)|j|jko)|j|jkWSt y4YdSwdS)NF) rrrrrr r r r r)rrrrr__eq__Ys$        zFingerprint.__eq__c Cs>ddd|jd|jd|jd|jd|jgS)N zSavedModel Fingerprintz saved_model_checksum: z graph_def_program_hash: z signature_def_hash: z saved_object_graph_hash: z checkpoint_hash: )joinrr r r r rrrr__str__gs     zFingerprint.__str__c Cs.d|jd|jd|jd|jd|jd S)Nz Fingerprint(z, ))rr r r r r!rrr__repr__qs zFingerprint.__repr__c Cshz t|j|j|j|jWSttjfy3}ztd|jd|jd|jd|j| dd}~ww)aCanonical fingerprinting ID for a SavedModel. Uniquely identifies a SavedModel based on the regularized fingerprint attributes. (saved_model_checksum is sensitive to immaterial changes and thus non-deterministic.) Returns: The string concatenation of `graph_def_program_hash`, `signature_def_hash`, `saved_object_graph_hash`, and `checkpoint_hash` fingerprint attributes (separated by '/'). Raises: ValueError: If the fingerprint fields cannot be used to construct the singleprint. z[Encounted invalid fingerprint values when constructing singleprint.graph_def_program_hash: zsignature_def_hash: zsaved_object_graph_hash: zcheckpoint_hash: N) fingerprinting_pywrap Singleprintr r r r TypeErrorFingerprintExceptionr)rrrrr singleprintxs0zFingerprint.singleprint)NNNNNN)__name__ __module__ __qualname____doc__intr classmethodrrrrboolrstrr"r$r%r&r)rrrrrs6  rz)saved_model.experimental.read_fingerprint export_dirrc Csvzt|}Wn)tjy}ztd|dd}~wtjy0}ztd|dd}~wwtt |S)a9Reads the fingerprint of a SavedModel in `export_dir`. Returns a `tf.saved_model.experimental.Fingerprint` object that contains the values of the SavedModel fingerprint, which is persisted on disk in the `fingerprint.pb` file in the `export_dir`. Read more about fingerprints in the SavedModel guide at https://www.tensorflow.org/guide/saved_model. Args: export_dir: The directory that contains the SavedModel. Returns: A `tf.saved_model.experimental.Fingerprint`. Raises: FileNotFoundError: If no or an invalid fingerprint is found. zSavedModel Fingerprint Error: N) r%ReadSavedModelFingerprintFileNotFoundExceptionFileNotFoundErrorr( RuntimeErrorrrrrr)r2 fingerprintrrrrread_fingerprints r8N) r-typingrtensorflow.core.protobufr0tensorflow.python.saved_model.pywrap_saved_modelrr% tensorflow.python.util.tf_exportrrr1r8rrrrs      z