o 2hy2@sddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZed GdddeZdS)N)backend)dtype_policies)tree) keras_export)any_symbolic_tensors)is_nnx_enabled)Node) KerasSaveable) python_utils)traceback_utils) auto_namezkeras.OperationcseZdZd&ddZejddZddZdd Zd d Z d d Z fddZ e j ddZeddZddZeddZeddZddZddZddZd d!Zd"d#Zd$d%ZZS)' OperationNcCsV|dur t|jj}t|trd|vr td|dt|d||_g|_g|_ dS)N/zRArgument `name` must be a string and cannot contain character `/`. Received: name=z (of type )) r __class____name__ isinstancestr ValueErrortypename_inbound_nodes_outbound_nodes)selfrrR/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/ops/operation.py__init__s  zOperation.__init__cOshtr[t||r |j}n;t|dddur9t|dddur+|j|jg|Ri|}n|j|jg|Ri|}nt|dddurE|j}n|j}tj||j j dd}||i|St||rh|j|i|St|dddurt|dddur|j|jg|Ri||i|S|j|jg|Ri||i|St|dddur|j|i|S|j|i|S)N _remat_modequantization_modez.call()) object_name) r is_traceback_filtering_enabledr symbolic_callgetattrrematerialized_callquantized_callcall!inject_argument_info_in_tracebackrr)rargskwargscall_fnrrr__call__ sd   zOperation.__call__cOs$|j|i|}t||||d|S)N) operation call_args call_kwargsoutputs)compute_output_specr)rr'r(r.rrrr!Os zOperation.symbolic_callcOtNNotImplementedErrorrr'r(rrrr%]zOperation.callcOr0r1r2r4rrrr$`r5zOperation.quantized_callcOszztj|jg|Ri|WSty<}z"|d|jd|jjd|jjd|jjd| }||jdd}~ww)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: ) rr/r% Exceptionrrrwith_traceback __traceback__)rr'r(enew_errrr/cs& zOperation.compute_output_specc sFtt||}tdkr5tr5ddlm}z |jt |d<Wnt y4|j t |d<Ynwt |jj}|tt|dt|d|tttttdf}zt|}d}|D] } t| |sld }nqaWn tyyd }Ynwzd |_|rdd lm } | j!d i||_"nd|_"d|_W|St#yY|Sw) 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__stateNTF)serialization_libr)$superr __new__rrflaxr< pytreelib PytreeStatevarsAttributeErrorobject ObjectStateinspectgetfullargspecrr'updatedictziplenrintfloatboolrrflattenr TypeError_lockkeras.src.savingr@SerializableDict _auto_configRecursionError) clsr'r(instancer< arg_namessupported_typesflat_arg_values auto_configvaluer@rrrrBrsJ  $      zOperation.__new__cCsd|ji}t|js |St|dddur@||jjt |j j }d|v}d|vo3|dj tj jk}|s>|s>|dd|Sttd|jjdddd ) zReturns the config of the object. An object config is a Python dictionary (serializable) containing the information needed to re-instantiate it. rrXNr(z Object 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: 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(arg1zC self.arg1, "arg2": self.arg2, z) return config)rr is_default get_configr"rLrXconfigrJ signaturer parameterskind Parameter VAR_KEYWORDpopr3textwrapdedentrr)rre init_params init_has_nameinit_has_kwargsrrrrds.  zOperation.get_configc Csd|vr(t|dtr(|}t|d}t|tjs$|jdur$|j}||d<z|di|WStyJ}zt d|j d|d|d}~ww)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) rrMcopyr deserializeDTypePolicyMaprrr6rTr)rZrepolicyr9rrr from_configs(   zOperation.from_configcCsd|jdS)Nz)rrrrr__repr__szOperation.__repr__cC|dddS)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_indexrxrrrr|  zOperation.inputcCrz)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_tensorsoutputr}rxrrrrrzOperation.outputcCs|jstd|jd|dt|j|ks'td|d|dt|jdt|j||}t|tr>t|dkr>|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.r?r)rrGrrOrr"rlist)r node_indexattr attr_namevaluesrrrr~!s$ z&Operation._get_node_attribute_at_indexcCdS)Nr rrxrrr _obj_typeBr5zOperation._obj_typecCr)5Can be overridden for per backend post build actions.Nrrxrrr _post_buildFzOperation._post_buildcCs||fS)rr)rrr`rrr _setattr_hookJszOperation._setattr_hookcCr)z5Can be overridden for per backend post track actions.Nrrvariablerrr_post_track_variableNrzOperation._post_track_variablecCr)z7Can be overridden for per backend post untrack actions.Nrrrrr_post_untrack_variableRrz Operation._post_untrack_variabler1)r __module__ __qualname__rr filter_tracebackr*r!r%r$r/rBr defaultrd classmethodrvrypropertyr|rr~rrrrr __classcell__rrrarr s0  . 3 6 )   !r )rJrl keras.srcrrrkeras.src.api_exportr%keras.src.backend.common.keras_tensorrkeras.src.backend.configrkeras.src.ops.noderkeras.src.saving.keras_saveabler keras.src.utilsr r keras.src.utils.namingr r rrrrs