o ?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_abcZ_default_save_signature signaturescCs2t|tjr|jdur|}t|tjsdS|SN) isinstancerFunctionZinput_signature(_get_concrete_function_garbage_collecteddefunConcreteFunctionrrv/home/www/facesmatcher.com/pyenv/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.FZunusedT)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 Z 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"rrrrBs  rc 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.Z_user_specified_name) zip function_def signatureZ input_arggraphinputsras_stropZget_attrr+r!)r#Z name_changesZsignature_input_nameZ graph_inputZuser_specified_namerrr_get_signature_name_changesOs      r3cCs||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)Z saveable_viewchildrenZpossible_signaturesr+childZconcreteZsingle_functionr.rrrfind_function_to_exportcs"     r<c 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.r4zrFunction `%s` contains input name(s) %s with unsupported characters which will be renamed to %s in the SavedModel.z, csdi|}t|jS)Nr)rr+)kwargsr Zsignature_function signature_keyrrsignature_wrappersz2canonicalize_signatures..signature_wrapper__name__Zsignature_wrapper_)experimental_attributescSs t|tjSr)rr TensorSpec)xrrrs z)canonicalize_signatures..T)Zexpand_composites)r+r)7rrMappingr Z!DEFAULT_SERVING_SIGNATURE_DEF_KEYitemsrr!getr Zwrap_cached_variablesr"_NUM_DISPLAY_NORMALIZED_SIGNATURESr3rinforr1r/r+joinkeysvalueshasattrrArZPOLYMORPHIC_FUNCTION_ALLOWLISTr-attrZ NO_INLINErrr*filterrr)r0r, _arg_keywordsrrCshapeZdtypeZ from_tensorrr9_num_positional_argsrZGenericFunctionZ function_specZ fullargspecdefaultslistTensor setdefault)rZ!num_normalized_signatures_counterZconcrete_signaturesZwrapped_functionsrTrZoriginal_functionZsignature_name_changesr@rBrOZ attr_valueZwrapped_functionZtensor_spec_signaturer0keywordr&specZfinal_concreteZflattened_defaultsZ len_default arg_namesargdefaultrr>rcanonicalize_signatures~s                     r]c 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%Z 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.)rrrFrNr^Sequence enumeraterGrZbytes_or_text_typesr!Z as_str_anyrrVrZCompositeTensor)rZ function_namer?keyvaluerrrrs6     rc@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.Nrh)rkr+r#rrr_add_signature'sz_SignatureMap._add_signaturecCs |j|Srrh)rkrerrr __getitem__-rmz_SignatureMap.__getitem__cC t|jSr)iterrirjrrr__iter__0rmz_SignatureMap.__iter__cCrpr)r9rirjrrr__len__3rmz_SignatureMap.__len__cCs d|jS)Nz_SignatureMap({}))r_rirjrrr__repr__6s z_SignatureMap.__repr__cKs"|tjjkriSdd|DS)NcSs(i|]\}}t|tjtjfr||qSr)rrrrr)r%rerfrrrra=s z5_SignatureMap._trackable_children..)r SaveTypeZ SAVEDMODELrG)rk save_typer=rrr_trackable_children9s z!_SignatureMap._trackable_childrenN)rA __module__ __qualname____doc__rlrnrorrrsrtr ru CHECKPOINTrwrrrrrg!srg signature_mapcCs t|tSr)rrg)objrrrrEFs rEcCstSr)rg)protorrrrELsr4)Zobject_factoryversionZmin_producer_versionZmin_consumer_versionsetter)versionscCsvt}|D]1\}}t|tjsJt|jtjsJt|j dkr+d|j ks*Jnd|j ks2J| ||q|S)z*Creates an object containing `signatures`.r4r) rgrGrrrr rrFr9rQrSrn)rr|r+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)r5r6SIGNATURE_ATTRIBUTE_NAMErrgr!)Zaugmented_graph_viewr+deprrrvalidate_augmented_graph_viewjs   rN)-rzZabslrZtensorflow.python.eagerrrrZ,tensorflow.python.eager.polymorphic_functionrZtensorflow.python.frameworkrrZtensorflow.python.opsr Ztensorflow.python.saved_modelr r r Ztensorflow.python.trackabler Ztensorflow.python.typesrZtensorflow.python.utilrrZtensorflow.python.util.compatrr7rrIrr$rr3r<r]rrFZ TrackablergZregister_revived_typeZVersionedTypeRegistrationrnrrrrrrsR                 z)#