&VfdZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z dZGddejZd Zd Z dd Z dd ZdZdS)zJSON encoder and decoder that handles TensorShapes and tuples.ctjrz _encode_tuple..$s*77=++777777rr c,g|]}t|Sr'r(s r z!_encode_tuple..'s ,,,Q a  ,,,rc4i|]\}}|t|Sr-r')r)keyvalues r z!_encode_tuple..)s&FFFjc5]5))FFFr)rtuplelistdictr )xs rrr s!U %77Q77777    At  ,,!,,,, At  FFAGGIIFFFFrc8tj|tS)N object_hook)jsonloads_decode_helper) json_strings rdecoder>.s :k~ > > >>rc dtj|tjtd||S)zFDecodes the JSON and deserializes any Keras objects found in the dict.T) deserializemodule_objectscustom_objectsr8)r:r; functoolspartialr<)r=rArBs rdecode_and_deserializerE2sA :% ))       rFc t|trd|vrtjr|ddkrtj|dS|ddkrGddlm}||dt|dS|dd kr|d }g}|d D]K\}}| tj |tj | Ltjt||d S|ddkrt!d|dDS|ddkrt"S|rGd|vrC d|vrt%j|||St)j|||S#t*$rYn+wxYw|ddkr|ddS|S)aJA decoding helper that is TF-object aware. Args: obj: A decoded dictionary that may represent an object. deserialize: Boolean. When True, deserializes any Keras objects found in `obj`. Defaults to `False`. module_objects: A dictionary of built-in objects to look the name up in. Generally, `module_objects` is provided by midlevel library implementers. custom_objects: A dictionary of custom objects to look the name up in. Generally, `custom_objects` is provided by the end user. Returns: The decoded object. r r r TypeSpecrtype_spec_registry type_spec serializedCompositeTensorspectensors)dtypeTexpand_compositesr%c34K|]}t|VdSr)r<r(s rr+z!_decode_helper..is*AAq**AAAAAAr __ellipsis____passive_serialization__module)rArB __bytes__r1utf-8)rr5rrr tensorflow.python.frameworkrIlookup _deserializer<appendconstantdtypesas_dtypenestpack_sequence_asr3Ellipsisrdeserialize_keras_objectr ValueErrorr) rr@rArBrIrMrNrOtensors rr<r<AsR$#t.0!4!4 < < M11~c'l333\"j00JJJJJJ)00[1ABBOO"3|#455\"&7776{%(^ME6NN F")2D2DU2K2KLLLw//"4(('T0 |  + +AACLAAAAA A  . 0 0O  08C?? C''(A'5'5 -E'5'5       + - -w<&&w// / JsF2F22 F?>F?c t|drtj|}d|d<|St|jt jkrBt|t jr| S| St|r|jSt|jtjkr|jStj r+t|tjjjr|jStj r.t|tjr|Stj r!t|tjr|jSt|t,jjrt3|S|t4urddiStj rt|tjrrddlm} |t|}d||d S#t@$r$tAd |d t|d wxYwtj rt|tj!j"rtj#|}g}tj$%|d D]G}|&|j'j|( fHdtS||dSt|tTj+r|jSt|tXrd|-ddSt]d |dt|d)zSerializes any object to a JSON-serializable structure. Args: obj: the object to serialize Returns: JSON-serializable structure representing `obj`. Raises: TypeError: if `obj` cannot be serialized. get_configTrTr rSrrHrG)r rJrKzUnable to serialize z% to JSON, because the TypeSpec class z has not been registered.rPrL)r rMrNrVrW)r r1z to JSON. Unrecognized type .)/hasattrrserialize_keras_objecttyper nprrndarraytolistitemcallablerrcompatv1 Dimensionr1r rDTypename collectionsabcMappingr5rarGrXrIget_name _serializerc __internal__rLtype_spec_from_valuer_flattenr[rOnumpyrenumEnumbytesr> TypeError)rrKrItype_spec_namerMrNrds rrrsVsL!!"9#>> 26 ./ Cyyr{** c2: & & ::<< 88:: }}| CyyT]**| | 3 (>??y | 377{{}} | 311x#{.//Cyy hn--  | 3 44BBBBBB /88cCCN(+!nn..     >s>>c>>>    |  3(GHH  &s++goocToBB I IF NNFL-v||~~/D/D/F/FG H H H H+!$''   #ty!!y#uI)CJJw4G4GHHH LsLLS LLL  s 9H.H=)NN)FNN)r"rur~rCr:r}rkkeras.src.legacy.savingrkeras.src.savingrkeras.src.utils.module_utilsrrr JSONEncoderrrr>rEr<rr-rrrsBB  111111......999999- 2 2 2 2 2d 2 2 2   ??? 6:     AEAAAAHWWWWWr