o ?ep @sdZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZeee e e e e eeeef ZefZeaddZddZdddZdS)z/Layer serialization/deserialization functions. N)tf2) base_layer) input_layer) input_spec)advanced_activations) convolutional)convolutional_recurrent)core)dense_attention) embeddings)merge)pooling) recurrent)rnn_cell_wrapper_v2) generic_utils) tf_inspectcs,ttds it_dt_tjrtjtkrdSit_tt_tjtj tjt fdddtr@tj tjt fdddddl m }tjtjd <tjtjd <|jtjd <|jtjd <|jtjd <tjtjd<tjtjd<tjtjd<tjtjd<tjtjd<tjtjd<tjtjd<tjtjd<dS)z8Populates dict ALL_OBJECTS with every built-in layer. ALL_OBJECTSNct|o t|SNinspectisclass issubclassxZbase_clsm/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/keras/layers/serialization.pyDz1populate_deserializable_objects..)Z obj_filtercrrrrrrrrKrrmodelsInput InputSpec FunctionalModel Sequentialaddsubtractmultiplyaveragemaximumminimum concatenatedot)hasattrLOCALrZGENERATED_WITH_V2renabledrZLayerrZ!populate_dict_with_module_objects ALL_MODULESALL_V2_MODULESZtensorflow.python.kerasr!rr"rr#r$r%r&r r'r(r)r*r+r,r-r.r rrrpopulate_deserializable_objects0sB                 r4cCs t|Sr)rZserialize_keras_object)layerrrr serializeas r6cCsttj|tj|ddS)aJInstantiates a layer from a config dictionary. Args: config: dict of the form {'class_name': str, 'config': dict} custom_objects: dict mapping class names (or function names) of custom (non-Keras) objects to class/functions Returns: Layer instance (may be Model, Sequential, Network, Layer...) r5)Zmodule_objectscustom_objectsZprintable_module_name)r4rZdeserialize_keras_objectr0r)configr7rrr deserializees r9r)__doc__ threadingZtensorflow.pythonrZtensorflow.python.keras.enginerrrZtensorflow.python.keras.layersrrrr r r r r rrZtensorflow.python.keras.utilsrrrr2r3localr0r4r6r9rrrrs6                1