2Vhn2ddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l mZdd lmZed Gd d Zy)N)backend)dtype_policies)tree) keras_export)any_symbolic_tensors)Node) python_utils)traceback_utils) auto_namezkeras.OperationceZdZddZej dZdZdZdZ dZ fdZ e jdZed Zd Zed Zed Zd ZdZdZdZdZxZS) Operationc|t|jj}t|trd|vrt d|dt |dtj||_ ||_ g|_ g|_ y)N/zRArgument `name` must be a string and cannot contain character `/`. Received: name=z (of type )) r __class____name__ isinstancestr ValueErrortyperget _dtype_policyname_inbound_nodes_outbound_nodes)selfdtypers G/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/ops/operation.py__init__zOperation.__init__s <T^^445D$$t ""&z$t*Q@  ,//6  !c6tjrt||r |j}nt |ddOt |dd!|j |j g|i|}nG|j |jg|i|}n&t |dd |j }n |j}tj||jjd}||i|St||r|j|i|St |ddYt |dd&|j |j g|i||i|S|j |jg|i||i|St |dd|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_fns r__call__zOperation.__call__s  9 9 ;#D&1,,45At%8$?K":$":": //#!#%# #;$":": II#(,#06#t%8$?K"&"5"5"&))%GG $ 7 78@GD+F+ + f -%4%%t6v6 6 T=$ / ;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)rr,r-r4s rr&zOperation.symbolic_callNs3*$**D;F; d r ctNNotImplementedErrorrr,r-s rr*zOperation.call\!!r ctr7r8r:s rr)zOperation.quantized_call_r;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: ) rr5r* Exceptionrrrwith_traceback __traceback__)rr,r-enew_es rr5zOperation.compute_output_specbs B..tyyJ4J6J J BKKII;k$..*A*A)BC#~~6678NN++,-))* -E&&q7T A Bs"% B5BB00B5c tt| |}tj|j j }|jtt|dt|dz||jdd}|Nt|tjr4|j|j |d<ntj"||d<t$t&t(t*t-df} t/j0|}d}|D]} t| |rd}n d|_|rddlm} | j:di||_nd|_d|_|S#t2$rd}YEwxYw#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()`. rNTFr)serialization_lib) superr __new__inspectgetfullargspecrr,updatedictziplenrrr DTypePolicyr#r serializerintfloatboolrrflatten TypeError_lockkeras.src.savingrESerializableDict _auto_configRecursionError) clsr,r-instance arg_namesrsupported_typesflat_arg_values auto_configvaluerErs rrHzOperation.__new__qsxC05**3<<8==  d3ySY];TBCD 7D)  E>3M3M!N&&."'**w#1":":5"AwUD$t*= "ll62OK( !%9"'K   "HN>(J(9(J(J))%)-%!HN % K     s*4(EE#5E+ E('E(+ E87E8c2d|ji}tj|js|St |ddt |j }|j|jjtj|j}|jdk7r8||j|jddz D]}|j!|d|St#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. rrYNr-rDz 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({ "arg1": self.arg1, "arg2": self.arg2, }) return config)rr is_default get_configr'setkeysrKrYconfigrIrJrvarkw intersectionr,popr9textwrapdedentrr)rrg xtra_argsargspeckeys rrdzOperation.get_configs DII &&t7M 4 . :FKKM*I MM$++22 3,,T]];G}}($y'='=gll12>N'OO*CJJsD)*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. rNz Error when deserializing class 'z' using config=z. Exception encountered: rF) rrLcopyr deserializeDTypePolicyMapr#rr>rUr)r[rgpolicyrAs r from_configzOperation.from_configs. f F7OT!B[[]F#//w@Fv~'D'DE,,4 $F7O ==  23<<.A !=aSB  s8B B, B''B,c"d|jdS)Nz)rrs r__repr__zOperation.__repr__s!$))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_indexrxs rr|zOperation.input s00OWMMr 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_tensorsoutputr}rxs rrzOperation.outputs004DhOOr 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.rDr)rAttributeErrorrrNrr'rlist)r node_indexattr attr_namevaluess rr~z&Operation._get_node_attribute_at_index"s "" TYYK(++4+Q8 4&&'*4 {),;t**+,O=  ,,Z8$? fd #F q(8!9 Mr cy)5Can be overridden for per backend post build actions.NrFrxs r _post_buildzOperation._post_buildD r c ||fS)rrF)rrras r _setattr_hookzOperation._setattr_hookHs U{r cy)z5Can be overridden for per backend post track actions.NrFrvariables r_post_track_variablezOperation._post_track_variableLrr cy)z7Can be overridden for per backend post untrack actions.NrFrs r_post_untrack_variablez Operation._post_untrack_variablePrr )NN)r __module__ __qualname__rr filter_tracebackr/r&r*r)r5rHr defaultrd classmethodrurypropertyr|rr~rrrr __classcell__)rs@rr r s "%%,2&,2\ "" B5n22h''R/ N N P PD   r r )rIrk keras.srcrrrkeras.src.api_exportr%keras.src.backend.common.keras_tensorrkeras.src.ops.noderkeras.src.utilsr r keras.src.utils.namingr r rFr rrsH$-F#(+, B B !B r