BVhuP.dZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z dd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"dZ#dZ$e"dgdZ%d+dZ&ejNfdZ(dZ)dZ*d+dZ+d+dZ,e"gd e jZd!d"Z.e"d#gd$Z/e"d%d&ge j`dd'd+d(Z1Gd)d*e2Z3y),zNLoader implementation for SavedModel with hermetic, language-neutral exports. N)message) text_format)graph_debug_info_pb2)meta_graph_pb2)saved_model_pb2)ops)file_io) variables) tf_logging) constants) path_helpers)signature_def_utils) utils_impl)metrics)saver)compat) deprecation) tf_exportloaderc t|}tjtj|t j }tj}tj|rDtj|d5} |j|j ddd||fS||fS#tj$r}td|d|dd}~wwxYw#1swY||fSxYw)a`Reads the savedmodel as well as the graph debug info. Args: export_dir: Directory containing the SavedModel and GraphDebugInfo files. Returns: `SavedModel` and `GraphDebugInfo` protocol buffers. Raises: IOError: If the saved model file does not exist, or cannot be successfully parsed. Missing graph debug info file is fine. rbCannot parse file : .N)parse_saved_modelr joinr get_debug_dirr DEBUG_INFO_FILENAME_PBrGraphDebugInfo file_existsFileIOParseFromStringreadr DecodeErrorIOError) export_dir saved_modeldebug_info_path debug_info debug_filees Y/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/saved_model/loader_impl.py!parse_saved_model_with_debug_infor-.s"*-+LL  ,&&(/$224* )  .D*D"":??#45D z ""+z "" D*?*;2aSBCCDD z ""s*C!B44CCCC!!C-z*__internal__.saved_model.parse_saved_model)v1c tjtjtj|tjt j }tjtjtj|tjt j}tjtjtj|tjt j}tj}tj|rCtj|d5}|j}ddd |j|Stj|rWtj|d5}|j}ddd t'j(j+d||St#d|t.j0j2dt j d t jd #1swYxYw#tj $r!}t#d|dt%|d|d}~wwxYw#1swYxYw#t&j,$r!}t#d|dt%|d|d}~wwxYw) a9Reads the savedmodel.pb or savedmodel.pbtxt file containing `SavedModel`. Args: export_dir: String or Pathlike, path to the directory containing the SavedModel file. Returns: A `SavedModel` protocol buffer. Raises: IOError: If the file does not exist, or cannot be successfully parsed. rNrrrzutf-8z#SavedModel file does not exist at: {|})r rras_bytes path_to_strr SAVED_MODEL_FILENAME_PBTXTSAVED_MODEL_FILENAME_PBSAVED_MODEL_FILENAME_CPBr SavedModelr r!r#r"rr$r%strrParsedecode ParseErrorospathsep)r& path_to_pbtxt path_to_pb path_to_cpbr'f file_contentr+s r,rrKs9,, oof((45 ooi::;=-|| oof((45 ooi778:*  oof((45 ooi889;+  **,+ $  D )QVVXlI!!,/ =)  t ,VVXlL ++G4kB   -j\"''++ Y 1 1 2!  , , -R 1 22   I ( Bs1vha@ AqHI  ! !L (r#a&C D!KLsHH;1I/I> %J ;II;I66I;>J J>J99J>c|j}i}g}|jr |j}nvtj|vrd|tjjj }|D]8}t j}|j||j|:tjtj|tjtj} |D]j}|jj } |r|d| } tjtj| tj|j"|| <l|S)aGets the asset tensors, if defined in the meta graph def to load. Args: export_dir: Directory where the SavedModel is located. meta_graph_def_to_load: The meta graph def from the SavedModel to be loaded. import_scope: Optional `string` -- if specified, prepend this followed by '/' to all returned asset tensor names. Returns: A dictionary of asset tensors, keyed by the name of the asset tensor. The value in the map corresponds to the absolute path of the asset file. /)collection_defasset_file_defr ASSETS_KEYany_listvaluer AssetFileDefUnpackappendr rrr3ASSETS_DIRECTORY tensor_infonamefilename) r&meta_graph_def_to_load import_scoperGasset_tensor_dict asset_protosassets_any_protoasset_any_proto asset_protoassets_directory tensor_names r,get_asset_tensorsr\~s+*88.,**)88L~-%i&:&:;DDJJ+'"//1k[)+&' \\ ooj!6??93M3M#NP"/k))..K+[9k%,\\() ,,-&/k" / c|j}d}||vrZ||jj}t|dk7rt dt|d|dt j |d}|S)aGets the main op tensor, if one exists. Args: meta_graph_def_to_load: The meta graph def from the SavedModel to be loaded. init_op_key: name of the collection to check; should be one of MAIN_OP_KEY or the deprecated LEGACY_INIT_OP_KEY Returns: The main op tensor, if it exists and `None` otherwise. Raises: RuntimeError: If the collection def corresponding to the main op key has other than exactly one tensor. Nz/Expected exactly one SavedModel init op. Found rrr)rG node_listrKlen RuntimeErrorrget_collection)rS init_op_keyrGinit_op init_op_lists r,_get_main_op_tensorrgs *88. 'N"!+.88>>L <A ""%l"3!4B|nAG HH  -a0G .r]ct||SN)rg)meta_graph_defop_keys r,_get_op_from_collectionrls ^V 44r]ch||jvr$tj|j|||Sy)z>Retrieve op stored in the imported meta graph's signature def.N) signature_defrload_op_from_signature_def)rjop_signature_keyrTs r,_get_op_from_signature_defrqs=555  9 9$$%568H  r]ct|tj|xs6t|tjxst|tj Sri)rqr INIT_OP_SIGNATURE_KEYrl MAIN_OP_KEYLEGACY_INIT_OP_KEY)rjrTs r, get_init_oprvsL $i55| E P !.)2G2G H P ".)2N2N OQr]ctt|tj|}|t|tj}|Sri)rqr TRAIN_OP_SIGNATURE_KEYrl TRAIN_OP_KEY)rjrTtrain_ops r, get_train_opr{s7 'i66 F( &~y7M7MNH /r]) saved_model.contains_saved_modelz'saved_model.maybe_saved_model_directory.saved_model.loader.maybe_saved_model_directoryr}c`tj|tj}tj|tj}tj|tj }tj |xs,tj |xstj |S)a8Checks whether the provided export directory could contain a SavedModel. Note that the method does not load any data by itself. If the method returns `false`, the export directory definitely does not contain a SavedModel. If the method returns `true`, the export directory may contain a SavedModel but provides no guarantee that it can be loaded. Args: export_dir: Absolute string path to possible export location. For example, '/my/foo/model'. Returns: True if the export directory contains SavedModel files, False otherwise. )r rr r5r6r7r )r&txt_pathpb_pathcpb_paths r,maybe_saved_model_directoryrs,\\*i&J&J K( LLY%F%F G' \\*i&H&H I( (#'   W %'   X &r]r|cvt|tjrtj|}t |S)a1Checks whether the provided export directory could contain a SavedModel. Note that the method does not load any data by itself. If the method returns `false`, the export directory definitely does not contain a SavedModel. If the method returns `true`, the export directory may contain a SavedModel but provides no guarantee that it can be loaded. Args: export_dir: Absolute path to possible export location. For example, '/my/foo/model'. Returns: True if the export directory contains SavedModel files, False otherwise. ) isinstancer=PathLikefspathr)r&s r,contains_saved_modelrs+  BKK(:&J $Z 00r]zsaved_model.loadzsaved_model.loader.loadz"Use `tf.saved_model.load` instead.c Bt|}|j|||fi|S)ax Loads the model from a SavedModel as specified by tags. Args: sess: The TensorFlow session to restore the variables. tags: Set of string tags to identify the required MetaGraphDef. These should correspond to the tags used when saving the variables using the SavedModel `save()` API. export_dir: Directory in which the SavedModel protocol buffer and variables to be loaded are located. import_scope: Optional `string` -- if specified, prepend this string followed by '/' to all loaded tensor names. This scope is applied to tensor instances loaded into the passed session, but it is *not* written through to the static `MetaGraphDef` protocol buffer that is returned. **saver_kwargs: Optional keyword arguments passed through to Saver. Returns: The `MetaGraphDef` protocol buffer loaded in the provided session. This can be used to further extract signature-defs, collection-defs, etc. Raises: RuntimeError: MetaGraphDef associated with the tags cannot be found. @compatibility(TF2) `tf.compat.v1.saved_model.load` or `tf.compat.v1.saved_model.loader.load` is not compatible with eager execution. Please use `tf.saved_model.load` instead to load your model. You can refer to the [SavedModel guide] (https://www.tensorflow.org/guide/saved_model) for more information as well as "Importing SavedModels from TensorFlow 1.x" in the [`tf.saved_model.load`] (https://www.tensorflow.org/api_docs/python/tf/saved_model/load) docstring. #### How to Map Arguments | TF1 Arg Name | TF2 Arg Name | Note | | :-------------------- | :-------------- | :------------------------- | | `sess` | Not supported | - | | `tags` | `tags` | - | | `export_dir` | `export_dir` | - | | `import_scope` | Not supported | Name scopes are not needed. : : : By default, variables are : : : : associated with the loaded : : : : object and function names : : : : are deduped. : | `saver_kwargs` | Not supported | - | #### Before & After Usage Example Before: ``` with tf.compat.v1.Session(graph=tf.Graph()) as sess: tf.compat.v1.saved_model.loader.load(sess, ["foo-tag"], export_dir) ``` After: ``` model = tf.saved_model.load(export_dir, tags=["foo-tag"]) ``` @end_compatibility )SavedModelLoaderload)sesstagsr&rT saver_kwargsrs r,rrs)D J '& T4 > >>r]cleZdZdZdZedZedZedZdZ d dZ d d Z d d Z d d Z y) rzrs; #':34+,+136=ID8).6 #: 7B?/@/d(X)2(=(=65Q  "!!466 2 -"5161( !#<=>(*??*???Dhvhr]