o i e2< @sRdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZdZdZdZddZddZddZ ddZ!ddZ"ddZ#d d!Z$Gd"d#d#ej%ej&Z'ej(d$d%d&ej)d'd&d(d(d(e'j*d)gd*d+d,Z+d-d.Z,d/S)0z;Helpers for working with signatures in tf.saved_model.save.)logging) def_functionfunction) attributes)composite_tensor)tensor)resource_variable_ops)function_serialization) revived_types)signature_constants)base)core)compat)nest)collections_abc_default_save_signature signaturescCs2t|tjr|jdur|}t|tjsdS|SN) isinstancerFunctioninput_signature(_get_concrete_function_garbage_collecteddefunConcreteFunctionrrd/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/saved_model/signature_serialization.py_get_signature(s   rcCs<|jsdSzt|t|jddWdStyYdSw)zBReturns whether concrete function can be converted to a signature.FunusedT)outputs_validate_inputs_normalize_outputsstructured_outputs ValueErrorconcrete_functionrrr_valid_signature3s r'cCs0tddt|jDrtd|jddS)z*Raises error if input type is tf.Variable.css|] }t|tjVqdSr)rr VariableSpec).0inprrr Ds  z#_validate_inputs..z'Unable to serialize concrete_function 'zc'with tf.Variable input. Functions that expect tf.Variable inputs cannot be exported as signatures.N)anyrflattenstructured_input_signaturer$namer%rrrr!Bs  r!c Csbi}t|jjj|jjD]"\}}zt|j d}|j |kr$|j ||<Wq t y.Yq w|S)zDChecks for user-specified signature input names that are normalized._user_specified_name) zip function_def signature input_arggraphinputsras_stropget_attrr/r$)r& name_changessignature_input_name graph_inputuser_specified_namerrr_get_signature_name_changesOs      r>cCs||j}g}|D]'\}}t|tjtjfsq |tkr |St|}|dur1t |r1| |q t |dkrH|d}t|}|rHt |rH|SdS)z;Function to export, None if no suitable function was found.Nr) list_childrenrootrrrrrDEFAULT_SIGNATURE_ATTRrr'appendlen) saveable_viewchildrenpossible_signaturesr/childconcretesingle_functionr3rrrfind_function_to_exportcs"     rKc s|dur iiifSt|tjstj|i}d}i}i}i}|D],\}t|}dur7td|d||p@t |||<t |t krnt }|rn|d7}tdtjjd|d|fdd }t|d rd |j|_i} tjD]} jj| d} | tjkr| dur| | | <q| sd} tj|| d } i} j durt!d dt"j#j dd}nj$}t%j&|D]&\}}t|}t|t'j(rt'j(|j)|j*|d}nt'j(j+||d}|| |<q| j,di| }t-|j&dkrd|_.nd|_.||<t|t/j0rMt"#|j1j2j3}t-|pg}t4| }|dkrMt%|| d|p4gD]\}}t|t'j5sCq6||6i|<q6q |||fS)z>Converts `signatures` into a dictionary of concrete functions.NrzJExpected a TensorFlow function for which to generate a signature, but got z_. Only `tf.functions` with an input signature or concrete functions can be used as a signature.r?zrFunction `%s` contains input name(s) %s with unsupported characters which will be renamed to %s in the SavedModel.z, csdi|}t|jS)Nr)r"r/)kwargsr#signature_function signature_keyrrsignature_wrappersz2canonicalize_signatures..signature_wrapper__name__signature_wrapper_)experimental_attributescSs t|tjSr)rr TensorSpec)xrrrs z)canonicalize_signatures..T)expand_composites)r/r)7rrMappingr !DEFAULT_SERVING_SIGNATURE_DEF_KEYitemsrr$getr wrap_cached_variablesr!"_NUM_DISPLAY_NORMALIZED_SIGNATURESr>rinforr7r5r/joinkeysvalueshasattrrQrPOLYMORPHIC_FUNCTION_ALLOWLISTr2attr NO_INLINErrr.filterrr-r6r1 _arg_keywordsrrTshapedtype from_tensorrrD_num_positional_argsrGenericFunction function_spec fullargspecdefaultslistTensor setdefault)r!num_normalized_signatures_counterconcrete_signatureswrapped_functionsrororiginal_functionsignature_name_changesrPrSrd attr_valuewrapped_functiontensor_spec_signaturer6keywordr*specfinal_concreteflattened_defaults len_default arg_namesargdefaultrrMrcanonicalize_signatures~s                     rc Cst|tjs"t|dr|}nt|tjs|g}ddt|D}|D]8\}}t|tj sAt d|dt |d|dt|t j tjfs^t d|d |dt |d|d q&|S) z?Normalize outputs if necessary and check that they are tensors._asdictcSsi|] \}}d||qS)z output_{})format)r) output_indexoutputrrr s z&_normalize_outputs..z'Got a dictionary with a non-string key z in the output of the function z+ used to generate the SavedModel signature .zGot a non-Tensor value z for key z. Outputs for functions used as signatures must be a single Tensor, a sequence of Tensors, or a dictionary from string to Tensor.)rrrXrbrSequence enumeraterZrbytes_or_text_typesr$ as_str_anyrrqrCompositeTensor)r function_namerOkeyvaluerrrr"s6     r"c@sPeZdZdZddZddZddZdd Zd d Zd d Z e j j fddZ dS) _SignatureMapz&A collection of SavedModel signatures.cCs i|_dSr _signaturesselfrrr__init__$ z_SignatureMap.__init__cCs||j|<dS)z&Adds a signature to the _SignatureMap.Nr)rr/r&rrr_add_signature'sz_SignatureMap._add_signaturecCs |j|Srr)rrrrr __getitem__-rz_SignatureMap.__getitem__cC t|jSr)iterrrrrr__iter__0rz_SignatureMap.__iter__cCrr)rDrrrrr__len__3rz_SignatureMap.__len__cCs d|jS)Nz_SignatureMap({}))rrrrrr__repr__6s z_SignatureMap.__repr__cKs"|tjjkriSdd|DS)NcSs(i|]\}}t|tjtjfr||qSr)rrrrr)r)rrrrrr=s z5_SignatureMap._trackable_children..)r SaveType SAVEDMODELrZ)r save_typerLrrr_trackable_children9s z!_SignatureMap._trackable_childrenN)rQ __module__ __qualname____doc__rrrrrrr r CHECKPOINTrrrrrr!sr signature_mapcCs t|tSr)rr)objrrrrVFs rVcCstSr)r)protorrrrVLsr?)object_factoryversionmin_producer_versionmin_consumer_versionsetter)versionscCsvt}|D]1\}}t|tjsJt|jtjsJt|j dkr+d|j ks*Jnd|j ks2J| ||q|S)z*Creates an object containing `signatures`.r?r) rrZrrrr#rrXrDrgrkr)rrr/funcrrrcreate_signature_mapVsrc CsP||jD]\}}|tkr%t|ts"td|jdtdtddSqdS)zCPerforms signature-related sanity checks on `augmented_graph_view`.zExporting an object z which has an attribute named 'z'. This is a reserved attribute used to store SavedModel signatures in objects which come from `tf.saved_model.load`. Delete this attribute (e.g. `del obj.z1`) before saving if this shadowing is acceptable.N)r@rASIGNATURE_ATTRIBUTE_NAMErrr$)augmented_graph_viewr/deprrrvalidate_augmented_graph_viewjs   rN)-rabslrtensorflow.python.eagerrrr,tensorflow.python.eager.polymorphic_functionrtensorflow.python.frameworkrrtensorflow.python.opsr tensorflow.python.saved_modelr r r tensorflow.python.trackabler tensorflow.python.typesrtensorflow.python.utilrrtensorflow.python.util.compatrrBrr]rr'r!r>rKrr"rX Trackablerregister_revived_typeVersionedTypeRegistrationrrrrrrrsR                 z)#