2Vh ddlZddlmZddlmZddlmZGddZGddejdgd Z d Z y) N)tree) KerasTensor)SymbolicArgumentscPeZdZdZ ddZdZedZedZedZ y) NodeaA `Node` describes an operation `__call__()` event. A Keras Function is a DAG with `Node` instances as nodes, and `KerasTensor` instances as edges. Nodes aren't `Operation` instances, because a single operation could be called multiple times, which would result in graph cycles. A `__call__()` event involves input tensors (and other input arguments), the operation that was called, and the resulting output tensors. A `Node` will include all this information. Since a single `Operation` could be called multiple times, the `Node` instances are stored on operations as a list. Each time an operation is called, a node is added to `op._inbound_nodes`. Each time the output of an operation is used by another operation, a node is added to `op._outbound_nodes`. Every `KerasTensor` instance has a `KerasHistory` object attached, which tracks the `Node` that records the `__call__()` event that created the tensor. By recursively walking through `Node` instances via the `KerasHistory` metadata of `KerasTensor` instances, once can retrieve the entire DAG of a Keras Function. Args: operation: The Operation that was called in the `op.__call__()` event that this node represents. call_args: The positional arguments the operation was called with. call_kwargs: The keyword arguments the operation was called with. outputs: The output tensors of the `op.__call__()` call. Nc`||_t|i||_|gntj||_|j D]#}t |trtd|d|td|jjD}|s;|jjD]"}t|drtddd|_ $|jjj||jjD]6}|jj} | | j j|8|sRt#|jjdz } t%|j D]\} }t|| | |_ |jj |_y)Nz1All operation outputs must be tensors. Operation z- returned a non-tensor. Non-tensor received: c36K|]}|j ywN)record_history).0xs B/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/ops/node.py z Node.__init__..6s %&   s_keras_historyr operation node_index tensor_index)rr argumentsrflattenoutputs isinstancer ValueErrorany keras_tensorshasattr KerasHistoryr_inbound_nodesappend_outbound_nodeslen enumerateis_input) selfr call_args call_kwargsrr zero_historytensorkt inbound_opris r__init__z Node.__init__(s#*IEE$_r$,,w2G  Aa- !!* ,,,-30  *...*F*F  ..66 v'78,8"&11-F)  %%,,T2.... 8B**44J%**11$7 8 T^^::;a?J&t||4  6(4'JQ)%  !NN888 cNd|jjdt|dS)Nz)rnameidr%s r__repr__z Node.__repr__Ts&!$.."5"5!6eBtH:QGGr.c.|jjSr )rrr3s r input_tensorszNode.input_tensorsWs~~+++r.c|jSr )rr3s routput_tensorszNode.output_tensors[s ||r.cg}|jjD]O}|jj}|jj}|2|j |j |Q|S)zyThe parent `Node`s. Returns: all the `Node`s whose output this node immediately depends on. )rrrrrr r)r% node_depsr*oprs r parent_nodeszNode.parent_nodes_sl .... @B"",,B**55J~  !2!2:!>?  @ r.)NNN) __name__ __module__ __qualname____doc__r-r4propertyr6r8r<r.rrrsW@DH*9XH,,  r.rceZdZdZdZy)raTracks the Operation call that created a Tensor. During construction of Keras Functions, this metadata is added to each Tensor produced as the output of an Operation. This allows Keras to track how each Tensor was produced, and this information is later retraced by the `Function` class to reconstruct the Operations graph. Attributes: operation: The Operation instance that produced the Tensor. node_index: The specific call to the Operation that produced this Tensor. Operations can be called multiple times in order to share weights. A new node is created every time an Operation is called. The corresponding node that represents the call event that produced the Tensor can be found at `op._inbound_nodes[node_index]`. tensor_index: The output index for this Tensor. Always zero if the Operation that produced this Tensor only has one output. Nested structures of Tensors are deterministically assigned an index via `nest.flatten`. rBN)r=r>r?r@ __slots__rBr.rrros .Ir.rrct|dS)Nr)r)objs ris_keras_tensorrGs 3( ))r.) collections keras.srcrkeras.src.backendr keras.src.ops.symbolic_argumentsrr namedtuplerrGrBr.rrMsC)>ddNKC>*r.