BVh6< dZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$GddejJejLZ'ejPddejRdd d d e'jT!g"d#Z+d$Z,y%)&z;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 signaturesct|tjr|j|j }t|t j sy|SN) isinstancerFunctioninput_signature(_get_concrete_function_garbage_collecteddefunConcreteFunctionrs e/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/saved_model/signature_serialization.py_get_signaturer(sD<001  " " .@@BH He44 5  /c|jsy t|t|jddy#t$rYywxYw)zBReturns whether concrete function can be converted to a signature.FunusedT)outputs_validate_inputs_normalize_outputsstructured_outputs ValueErrorconcrete_functions r_valid_signaturer*3sJ  " " &'(;;XxP   s "2 >>ctdtj|jDrt d|j dy)z*Raises error if input type is tf.Variable.c3PK|]}t|tj ywr)rr VariableSpec).0inps r z#_validate_inputs..Ds&  +889 s$&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(s rr$r$BsU /JJK   12C2H2H1IJ3 3  r cNi}t|jjj|jj D]R\}} t j|jjd}|j|k7r|j||<T|S#t$rYbwxYw)zDChecks for user-specified signature input names that are normalized._user_specified_name) zip function_def signature input_arggraphinputsras_stropget_attrr4r')r) name_changessignature_input_name graph_inputuser_specified_names r_get_signature_name_changesrDOs,+.$$..88$$,  'K "MM .. ! !"8 9  " "&9 9,@,E,E ()       sA B B$#B$cr|j|j}g}|D]h\}}t|tjt j fs1|tk(r|cSt|}|Lt|sX|j|jt|dk(r|d}t|}|r t|r|Sy)z;Function to export, None if no suitable function was found.Nr) list_childrenrootrrrrrDEFAULT_SIGNATURE_ATTRrr*appendlen) saveable_viewchildrenpossible_signaturesr4childconcretesingle_functionr9s rfind_function_to_exportrRcs ( ();); <( +kdE el33U5K5KL M %% le$H 0 :  *+ ")!,O/I%i0  r c |iiifSt|tjstj|i}d}i}i}i}|j D]y\}t |x}td|d|j|xstj|x||<t|tkrt}|r~|dz }tjdt!j"j$j&dj)|j+dj)|j-fd}t/|d rd |j0z|_i} t2j4D]D} j6j8j| d} | t2j:k7s=| @| | | <F| sd} t=j>|| } i} j@-tCd tEjFj@d }n jH}tKjL|D]\}}t!j"|}t|tNjPr-tOjP|jR|jT|}n!tNjPjW||}|| |<| jXdi| }t[|jLdk(rd|_.nd|_.||<t|t^j`stEjF|jbjdjf}t[|xsg}ti| j+}|dkDs.tK|| d|xsgD]5\}}t|tNjjs!||jmi|<7||||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.rFzrFunction `%s` contains input name(s) %s with unsupported characters which will be renamed to %s in the SavedModel.z, cBdi|}t|jS)N)r%r4)kwargsr&signature_function signature_keys rsignature_wrapperz2canonicalize_signatures..signature_wrappers,-77  055}r __name__signature_wrapper_)experimental_attributesc6t|tjSr)rr TensorSpec)xs rz)canonicalize_signatures..sJq&"3"34r T)expand_composites)r4rU)7rrMappingr !DEFAULT_SERVING_SIGNATURE_DEF_KEYitemsrr'getr wrap_cached_variablesr$"_NUM_DISPLAY_NORMALIZED_SIGNATURESrDrinforr=r;r4joinkeysvalueshasattrrZrPOLYMORPHIC_FUNCTION_ALLOWLISTr8attr NO_INLINErrr3filterrr2r<r7 _arg_keywordsr r^shapedtype from_tensorrrK_num_positional_argsrPolymorphicFunction function_spec fullargspecdefaultslistTensor setdefault)r!num_normalized_signatures_counterconcrete_signatureswrapped_functionsryroriginal_functionsignature_name_changesrYr\rn attr_valuewrapped_functiontensor_spec_signaturer<keywordr/specfinal_concreteflattened_defaults len_default arg_namesargdefaultrWrXs @@rcanonicalize_signaturesr~s r2: J 7 7 8==zJ'(# (!+!1!1!3j@mX-;H-EE*!  J; ;  /0 K ! 7 78I J'(+='((+MM:;MN )Q.)  H MM,2277 8 II,113 4 II,335 6   x$#7(:K:K#K !993%2277;;D$Gj %% %**@(2%3 # $#,,3J44@ 4 ,, ;; $ f"((f(( ,  g&g C** +  CIIGD  ,,Sw,?'+G$ ,O%NN N > ' '(A- -.n),-n))7 &(D445<<  , , 5 5*0b1k,1134i q {lm $  $"  @LCGV]]3 8?(  mR 0 5  @Ij@V / 99r c rt|tjsgt|dr|j }nJt|tj s|g}t |Dcic]\}}dj||}}}|jD]\}}t|tjs(td|dtj|d|dt|tjtj frstd|d|dtj|d|d |Scc}}w) z?Normalize outputs if necessary and check that they are tensors._asdictz output_{}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.)rrrbrlrSequence enumerateformatrdrbytes_or_text_typesr' as_str_anyr r{rCompositeTensor)r# function_namerX output_indexoutputkeyvalues rr%r%sO G_44 5w "!g !9!9 :)'0&8"lF   \ *F 2g MMOjc5 c655 6  3C7; ++M:;<&&3%6a 9  efmm-=-M-MN O  #E9IcW= ++M:;<&&3%67J J  .+s#D3cfeZdZdZdZdZdZdZdZdZ e jjfdZ y ) _SignatureMapz&A collection of SavedModel signatures.ci|_yr _signaturesselfs r__init__z_SignatureMap.__init__$s Dr c"||j|<y)z&Adds a signature to the _SignatureMap.Nr)rr4r)s r_add_signaturez_SignatureMap._add_signature's/DTr c |j|Srr)rrs r __getitem__z_SignatureMap.__getitem__-s   C  r c,t|jSr)iterrrs r__iter__z_SignatureMap.__iter__0s   !!r c,t|jSr)rKrrs r__len__z_SignatureMap.__len__3s t  r c8dj|jS)Nz_SignatureMap({}))rrrs r__repr__z_SignatureMap.__repr__6s  % %d&6&6 77r c |tjjk7riS|jDcic]2\}}t |t j tjfr||4c}}Scc}}wr) rSaveType SAVEDMODELrdrrrrr)r save_typerVrrs r_trackable_childrenz!_SignatureMap._trackable_children9sdDMM,,, i**,  C el33U5K5KL M U   s7A.N)rZ __module__ __qualname____doc__rrrrrrrr CHECKPOINTrrUr rrr!s7./ !"!8+/--*B*Br r signature_mapc"t|tSr)rr)objs rr`r`Fs 3 .r ctSr)r)protos rr`r`Lsr rF)object_factoryversionmin_producer_versionmin_consumer_versionsetter)versionscft}|jD]\}}t|tjsJt|j t jsJt|jdk(rd|jk(sJd|jk(sJ|j|||S)z*Creates an object containing `signatures`.rFr) rrdrrrr&rrbrKrqrur)rrr4funcs rcreate_signature_maprVs/-$$& -jdD dE22 33 3 d--/F/F GG G 4  !# $++ ++ + $++ ++ +  t, - r c |j|jD]G\}}|tk(st|ts't d|jdtdtdyy)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)rGrHSIGNATURE_ATTRIBUTE_NAMErrr')augmented_graph_viewr4deps rvalidate_augmented_graph_viewrjs'55  idC '' ] +"#7#<#<"=>!!9 :;223(  (     r N)-rabslrtensorflow.python.eagerrrr,tensorflow.python.eager.polymorphic_functionrtensorflow.python.frameworkrr tensorflow.python.opsr tensorflow.python.saved_modelr r r tensorflow.python.trackablertensorflow.python.typesrtensorflow.python.utilrrtensorflow.python.util.compatrrIrrgrr*r$rDrRrr%rb Trackablerregister_revived_typeVersionedTypeRegistrationrrrrUr rrsB05C8.7@7=,()'92'%&"  (6w:tR O++T^^ F$ ##./ //9!"!" //  $( r