o bi}<@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)selfrrK/home/ubuntu/.local/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 stt||}tdkr5tr5ddlm}z |jt |d<Wnt y4|j t |d<Ynwd}t |j}t |jz|jdg|Ri|}Wn ty\d}Ynw|rrtfd d |jDrrd}d|_|r||j}||jiO}|jd|d djdur|j}t|dkrd}d|_ttt t!t"df} zt#$|} | D] } t%| | sd}nqWn tyd}Ynwzd|_&|rdd l'm(} | 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__stateTNFcs,g|]\}}|jdkr|jtjjkqS)r)r'kindinspect ParameterPOSITIONAL_ONLY).0rparamargspecrr s  z%Operation.__new__..r)serialization_libr),superr __new__rrflaxr< pytreelib PytreeStatevarsAttributeErrorobject ObjectStater@ signaturergetfullargspecbind TypeErrorany parametersitems_auto_config_error_argsapply_defaults argumentspopvarkwr'varargslenrintfloatboolrrflattenr_lockkeras.src.savingrHSerializableDict _auto_configRecursionError) clsr'r(instancer< auto_configrRbound_parametersr^supported_typesflat_arg_valuesvaluerHrrErrJrsx                 zOperation.__new__cCsd|ji}t|js |St|dddur@||jjt |j j }d|v}d|vo3|dj tj jk}|s>|s>|dd|Sd}t|ddrWttd|jjd |ttd|jjd |) zReturns the config of the object. An object config is a Python dictionary (serializable) containing the information needed to re-instantiate it. rrgNr(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 rYFz 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"updatergconfigr@rRrrWr?rA VAR_KEYWORDr\r3textwrapdedentrr)rrt init_params init_has_nameinit_has_kwargs example_strrrrrrsD    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) rdictcopyr deserializeDTypePolicyMaprrr6rUr)rirtpolicyr9rrr from_configs(   zOperation.from_configcCsd|jdS)Nz)rrrrr__repr__DszOperation.__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_indexrrrrrG zOperation.inputcCr)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_tensorsoutputrrrrrrSrzOperation.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)rrOrr_rr"rlist)r node_indexattr attr_namevaluesrrrr_s$ z&Operation._get_node_attribute_at_indexcCdS)Nr rrrrr _obj_typer5zOperation._obj_typecCr)5Can be overridden for per backend post build actions.Nrrrrr _post_buildzOperation._post_buildcCs||fS)rr)rrrorrr _setattr_hookszOperation._setattr_hookcCr)z5Can be overridden for per backend post track actions.Nrrvariablerrr_post_track_variablerzOperation._post_track_variablecCr)z7Can be overridden for per backend post untrack actions.Nrrrrr_post_untrack_variablerz Operation._post_untrack_variabler1)r __module__ __qualname__rr filter_tracebackr*r!r%r$r/rJr defaultrr classmethodrrpropertyrrrrrrrr __classcell__rrrprr s0  . [ L )   !r )r@rv 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