ijZAddlZddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZedGddeZy)N)backend)dtype_policies)tree) keras_export) KerasTensor)remat)any_symbolic_tensors)is_nnx_enabled)Node) KerasSaveable) python_utils)traceback_utils) auto_namezkeras.OperationceZdZddZej dZdZdZdZ dZ dZ fdZ e jd Zed Zd Zed Zed ZdZdZdZdZdZdZxZS) Operationc|t|jj}t|trd|vrt d|dt |d||_g|_g|_ tj|_ y)N/zRArgument `name` must be a string and cannot contain character `/`. Received: name=z (of type )) r __class____name__ isinstancestr ValueErrortypename_inbound_nodes_outbound_nodesrget_current_remat_mode _remat_mode)selfrs l/var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/keras/src/ops/operation.py__init__zOperation.__init__s~ <T^^445D$$t ""&z$t*Q@    ! 779c2tjrt||r |j}n|jrOt |dd!|j |jg|i|}nG|j |jg|i|}n&t |dd |j}n |j}tj||jjd}||i|St||r|j|i|S|jrYt |dd&|j |jg|i||i|S|j |jg|i||i|St |dd|j|i|S|j|i|S)Nquantization_modez.call()) object_name) ris_traceback_filtering_enabledr symbolic_callrgetattrrematerialized_callquantized_callcall!inject_argument_info_in_tracebackrr)r argskwargscall_fns r!__call__zOperation.__call__$s  9 9 ;#D&1,,##t%8$?K":$":": //#!#%# #;$":": II#(,#06#t%8$?K"&"5"5"&))%GG $ 7 78@GD+F+ + f -%4%%t6v6 6   t0$7C/t//''*.28#!##L/t// KDKFK#t0$7C*t**D;F;; tyy$1&11r#cH|j|i|}t|||||S)N) operation call_args call_kwargsoutputs)compute_output_specr )r r.r/r6s r!r(zOperation.symbolic_callSs3*$**D;F; d r#ctNNotImplementedErrorr r.r/s r!r,zOperation.calla!!r#ctr9r:r<s r!r+zOperation.quantized_calldr=r#c|js|Sd}|jjdk(rtj|S|jjdk(rt|j|i|}t t j t j||}|r.||jjkDrtj|S|S)zEnable rematerialization dynamically for an operation's call method. Args: fn: The original `call` or `quantized_call` method of an operation. Returns: Rematerialized method. ct|tr1tj|jDcgc]}|xsd c}SdScc}w)Nr)rrmathprodshape)xds r! compute_sizez3Operation.rematerialized_call..compute_sizessCa- 1773a16634  3s Afull larger_than) rmoderr7sumrflatten map_structureoutput_size_threshold)r fnr.r/rG output_spec output_sizes r!r*zOperation.rematerialized_callgsI     F *;;r? "   " "m 32$22DCFCK T// kJKK$"2"2"H"HH{{2& r#c p tj|jg|i|S#t$r}|j d|j d|jj d|jj d|jj d| }|j|jdd}~wwxYw)Nz;Could not automatically infer the output shape / dtype of 'z ' (of type z). Either the `z<.call()` method is incorrect, or you need to implement the `zM.compute_output_spec() / compute_output_shape()` method. Error encountered: ) rr7r, Exceptionrrrwith_traceback __traceback__)r r.r/enew_es r!r7zOperation.compute_output_specs B..tyyJ4J6J J BKKII;k$..*A*A)BC#~~6678NN++,-))* -E&&q7T A Bs"% B5BB00B5c tt| |}tjdk(r7t r-ddlm} |jjt|d<d}tj|j}tj |j} |j"dg|i|}|rvt'|j(j+D cgc]>\} } | |j,dk7r'| j.tj0j2k(@c} } r d}d|_|rj7|j8}||j;|j<iz}|j;|j,d|j;d d|j>2|j;|j>} tA| dkDr d}d|_tBtDtFtHtKdf} tMjN|} | D]}tQ|| rd}n d|_)|rdd l*m+}|jXd i||_-nd|_-d|_)|S#t$r*|jjt|d<Y1wxYw#t$$rd}YwxYwcc} } w#t$$rd}YwxYw#t\$rY|SwxYw) aJWe override __new__ to saving serializable constructor arguments. These arguments are used to auto-generate an object serialization config, which enables user-created subclasses to be serializable out of the box in most cases without forcing the user to manually implement `get_config()`. jaxr)nnx_pytree__state_object__stateTNFr)serialization_lib)/superr__new__rr flaxrZ pytreelib PytreeStatevarsAttributeErrorobject ObjectStateinspect signaturer"getfullargspecbind TypeErrorany parametersitemsr.kind ParameterPOSITIONAL_ONLY_auto_config_error_argsapply_defaults argumentspopvarkwvarargslenrintfloatboolrrrLr_lockkeras.src.savingr]SerializableDict _auto_configRecursionError)clsr.r/instancerZ auto_configriargspecbound_parametersrparamrxsupported_typesflat_arg_valuesvaluer]rs r!r`zOperation.__new__sC05 ??  %.*:  L36==3L3L3NX/0  %%cll3 ((6 -y~~dDTDVD  3$-#7#7#=#=#? D%7<<?* g//???   K/3H ,   + + -%//F fjj3 3F JJw||A ' JJvt $* **W__5w(J(9(J(J))%)-%!HN W" L36::3I3I3KX/0 L  K  P K     sT&I"$JAJ* >&J0%J0+5K"/JJ J'&J'0 J>=J> K Kcd|ji}tj|js|St |dd|j |j jtj|jj}d|v}d|vxr*|djtjjk(}|s|s|jdd|Sd}t |ddr8t!t#j$d|j&j(d |t!t#j$d|j&j(d |) zReturns the config of the object. An object config is a Python dictionary (serializable) containing the information needed to re-instantiate it. rrNr/a class CustomLayer(keras.layers.Layer): def __init__(self, arg1, arg2, **kwargs): super().__init__(**kwargs) self.arg1 = arg1 self.arg2 = arg2 def get_config(self): config = super().get_config() config.update({ "arg1": self.arg1, "arg2": self.arg2, }) return config rsFz Object a was created by passing positional only or variadic positional arguments (e.g., `*args`) to `__init__()`, which is not supported by the automatic config generation. Please remove all positional only and variadic arguments from `__init__()` or override `get_config()` and `from_config()` to make the object serializatble. Example: a was created by passing non-serializable argument values in `__init__()`, and therefore the object must override `get_config()` in order to be serializable. Please implement `get_config()`. Example: )rr is_default get_configr)updaterconfigrhrir"rnrprq VAR_KEYWORDrvr;textwrapdedentrr)r r init_params init_has_nameinit_has_kwargs example_strs r!rzOperation.get_configs\ DII &&t7M 4 . : MM$++22 3!++DMM:EEK"k1MK'P)..'2C2C2O2OO ! 64(MKt6>)OONN++, - M  "*OONN++,- M   r#c ^d|vrrt|dtr_|j}tj|d}t|tj s|j |j}||d< |di|S#t$r#}td|jd|d|d}~wwxYw)aCreates an operation from its config. This method is the reverse of `get_config`, capable of instantiating the same operation from the config dictionary. Note: If you override this method, you might receive a serialized dtype config, which is a `dict`. You can deserialize it as follows: ```python if "dtype" in config and isinstance(config["dtype"], dict): policy = dtype_policies.deserialize(config["dtype"]) ``` Args: config: A Python dictionary, typically the output of `get_config`. Returns: An operation instance. dtypeNz Error when deserializing class 'z' using config=z. Exception encountered: r^) rdictcopyr deserializeDTypePolicyMapr%rrSrlr)rrpolicyrVs r! from_configzOperation.from_configBs. f F7OT!B[[]F#//w@Fv~'D'DE,,4 $F7O ==  23<<.A !=aSB  s8B B, B''B,c"d|jdS)Nz)rr s r!__repr__zOperation.__repr__ls!$))A..r#c(|jdddS)zRetrieves the input tensor(s) of a symbolic operation. Only returns the tensor(s) corresponding to the *first time* the operation was called. Returns: Input tensor or list of input tensors. r input_tensorsinput_get_node_attribute_at_indexrs r!rzOperation.inputos00OWMMr#c(|jdddS)zRetrieves the output tensor(s) of a layer. Only returns the tensor(s) corresponding to the *first time* the operation was called. Returns: Output tensor or list of output tensors. routput_tensorsoutputrrs r!rzOperation.output{s004DhOOr#c N|jstd|jd|dt|j|kDs(t d|d|dt|jdt |j||}t |trt|dk(r|d S|S) aPrivate utility to retrieves an attribute (e.g. inputs) from a node. This is used to implement the properties: - output - input Args: node_index: Integer index of the node from which to retrieve the attribute. attr: Exact node attribute name. attr_name: Human-readable attribute name, for error messages. Returns: The operation's attribute `attr` at the node of index `node_index`. z The layer z/ has never been called and thus has no defined .z Asked to get z at node z, but the operation has only z inbound nodes.rAr)rrerryrr)rlist)r node_indexattr attr_namevaluess r!rz&Operation._get_node_attribute_at_indexs "" TYYK(++4+Q8 4&&'*4 {),;t**+,O=  ,,Z8$? fd #F q(8!9 Mr#cy)Nrr^rs r! _obj_typezOperation._obj_typesr#cy)5Can be overridden for per backend post build actions.Nr^rs r! _post_buildzOperation._post_build r#c ||fS)rr^)r rrs r! _setattr_hookzOperation._setattr_hooks U{r#cy)z5Can be overridden for per backend post track actions.Nr^r variables r!_post_track_variablezOperation._post_track_variablerr#cy)z7Can be overridden for per backend post untrack actions.Nr^rs r!_post_untrack_variablez Operation._post_untrack_variablerr#r9)r __module__ __qualname__r"rfilter_tracebackr1r(r,r+r*r7r`r defaultr classmethodrrpropertyrrrrrrrr __classcell__)rs@r!rrs :%%,2&,2\ """H BYvJJX''R/ N N P PB   r#r)rhrBr keras.srcrrrkeras.src.api_exportrkeras.src.backendrkeras.src.backend.commonr%keras.src.backend.common.keras_tensorr keras.src.backend.configr keras.src.ops.noder keras.src.saving.keras_saveabler keras.src.utilsr rkeras.src.utils.namingrrr^r#r!rsY $-)*F3#9(+, e e !e r#