BVhtdZddlmZddlmZddlmZddlmZddl m Z ddl m Z e dg Gd d e Zy ) zlSignatureDef method name utility functions. Utility functions for manipulating signature_def.method_names. )file_io) tf_logging) constants) loader_impl)compat) tf_exportz1saved_model.signature_def_utils.MethodNameUpdater)v1c&eZdZdZdZddZddZy)MethodNameUpdateraUpdates the method name(s) of the SavedModel stored in the given path. The `MethodNameUpdater` class provides the functionality to update the method name field in the signature_defs of the given SavedModel. For example, it can be used to replace the `predict` `method_name` to `regress`. Typical usages of the `MethodNameUpdater` ```python ... updater = tf.compat.v1.saved_model.signature_def_utils.MethodNameUpdater( export_dir) # Update all signature_defs with key "foo" in all meta graph defs. updater.replace_method_name(signature_key="foo", method_name="regress") # Update a single signature_def with key "bar" in the meta graph def with # tags ["serve"] updater.replace_method_name(signature_key="bar", method_name="classify", tags="serve") updater.save(new_export_dir) ``` Note: This function will only be available through the v1 compatibility library as tf.compat.v1.saved_model.builder.MethodNameUpdater. cF||_tj||_y)zCreates an MethodNameUpdater object. Args: export_dir: Directory containing the SavedModel files. Raises: IOError: If the saved model file does not exist, or cannot be successfully parsed. N) _export_dirloaderparse_saved_model _saved_model)self export_dirs a/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/saved_model/method_name_updater.py__init__zMethodNameUpdater.__init__8s"D00> ?Ad""4T->->)?@ \\ //. ) //);; <>d"" !!33$3GIOO/1EFr)N)__name__ __module__ __qualname____doc__rr#r3rrr r s0 =*NXGrr N)r7tensorflow.python.lib.iortensorflow.python.platformrtensorflow.python.saved_modelrrrtensorflow.python.utilr tensorflow.python.util.tf_exportrobjectr r8rrr?sI -13?)6  BCDpGpGEpGr