o 2h @sddlZddlZddlZddlZddlmZddlmZddl m Z ddZ dddZ Gd d d Z Gd d d Zed ddZdS)N)backend) keras_export) global_statecCs4tddrdSdtjvrddlm}| SdS)Nin_tf_graph_scopeFT tensorflowr)r)rget_global_attributesysmoduleskeras.src.utils.module_utilsrexecuting_eagerly)tfr X/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/utils/backend_utils.py in_tf_graph s    rcCs$tdkrtstj||d}|S)Nr)dtype)backend_modulerrconvert_to_tensor)outputsrr r rconvert_tf_tensorsrc@s$eZdZddZddZddZdS) TFGraphScopecCstdd|_dS)NrF)rr_original_valueselfr r r__init__s zTFGraphScope.__init__cCstdddS)NrT)rset_global_attributerr r r __enter__"szTFGraphScope.__enter__cOstd|jdS)Nr)rrr)rargskwargsr r r__exit__%szTFGraphScope.__exit__N)__name__ __module__ __qualname__rrrr r r rrs rc@s>eZdZdZd ddZddZddZed d Zd d Z dS)DynamicBackendaFA class that can be used to switch from one backend to another. Example: ```python backend = DynamicBackend("tensorflow") y = backend.square(tf.constant(...)) backend.set_backend("jax") y = backend.square(jax.numpy.array(...)) ``` Args: backend: Initial backend to use (string). NcCs|pt|_dSNrr_backendrrr r rr;szDynamicBackend.__init__cCs |dvr td|||_dS)N)rjaxtorchnumpyopenvinozaAvailable backends are ('tensorflow', 'jax', 'torch', 'numpy' and 'openvino'). Received: backend=) ValueErrorr%r&r r r set_backend>s zDynamicBackend.set_backendcCst|_dSr#r$rr r rresetFszDynamicBackend.resetcCs|jSr#)r%rr r rnameIszDynamicBackend.namecCs|jdkrtd}t||S|jdkrtd}t||S|jdkr-td}t||S|jdkrAtdkr=tt|Std|jd krPtd }t||SdS) Nrzkeras.src.backend.tensorflowr'zkeras.src.backend.jaxr(zkeras.src.backend.torchr)zoCurrently, we cannot dynamically import the numpy backend because it would disrupt the namespace of the import.r*zkeras.src.backend.openvino)r% importlib import_modulegetattrrrNotImplementedError)rr.moduler r r __getattr__Ms&               zDynamicBackend.__getattr__r#) rr r!__doc__rr,r-propertyr.r4r r r rr"+s   r"zkeras.config.set_backendcCs|tjd<ddtjD}|D]}tj|=qddl}tt}|D]1\}}|j |j krWt | drWt |}|| ddd}|d| d}t |t|<q&dS) uReload the backend (and the Keras package). Example: ```python keras.config.set_backend("jax") ``` ⚠️ WARNING ⚠️: Using this function is dangerous and should be done carefully. Changing the backend will **NOT** convert the type of any already-instantiated objects. Thus, any layers / tensors / etc. already created will no longer be usable without errors. It is strongly recommended **not** to keep around **any** Keras-originated objects instances created before calling `set_backend()`. This includes any function or class instance that uses any Keras functionality. All such code needs to be re-executed after calling `set_backend()`. KERAS_BACKENDcSsg|] }|dr|qS)keras) startswith).0keyr r r |s  zset_backend..rNzs    9