o ?e?,@sdZddlZddlZddlZddlZddlmZddlmZ ddlm Z ddl m Z ddl mZddlmZdd lmZdd lmZd ZGd d d ZGdddedgdZddZdS)zContains the `Node` class.N)ops)tensor) tensor_util)backend)base_layer_utils) json_utils)tf_utils)nest_CONSTANT_VALUEc@seZdZdZ   dddZeddZeddZd d Zd d Z d dZ eddZ eddZ eddZ eddZeddZeddZdS)Nodea A `Node` describes the connectivity between two layers. Each time a layer is connected to some new input, a node is added to `layer._inbound_nodes`. Each time the output of a layer is used by another layer, a node is added to `layer._outbound_nodes`. Args: layer: The Layer for the Layer.__call__ this node represents. call_args: The positional arguments the Layer was called with. call_kwargs: The keyword arguments the Layer was called with. outputs: The outputs of the Layer.__call__ NcCs|durgn|}|durin|}|durgn|}||_| o | |_tdd|}tdd|}tdd||_||_||_t|j|jf|_|j o]t |jdko]t |jd|_ t s{|jD]}t|tjrztj|ddrzt|qfg|_g|_t|jD]\}}t|r|j|tt|}|} |j|| fq|jj||jD]} | jj} | dur| j|qt |jjd} tt|D] \}} t || |d | _qd d |jD|_!d d t|jD|_"dS) NcS|SNtrrd/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/keras/engine/node.pyEzNode.__init__..cSr r rrrrrrFrcSr r rrrrrrGrrT)Zignore_call_contextlayer node_index tensor_indexcSg|]}tt|qSrstrid.0rrrr pz!Node.__init__..cSrrrrrrrrqr )#ris_inputr map_structureoutputs call_args call_kwargsflatten_flat_argumentslenrZ is_tf_type _single_positional_tensor_passedrZ#executing_eagerly_outside_functions isinstance tensor_libTensorrZneeds_keras_historyZcreate_keras_history _keras_inputs_keras_inputs_ids_and_indices enumerateis_keras_tensorappendrr_inbound_nodes keras_inputs_keras_historyZ_outbound_nodes KerasHistoryZflat_input_idsZflat_output_ids)selfrr$r%r#objiZelekt_idkt_indexktZ inbound_layerrrrrr__init__3sb         z Node.__init__cC|jS)zFTensors input to this node that can be traced back to a `keras.Input`.)r-r6rrrr3sszNode.keras_inputscCs<g}|jD]}|jj}|jj}|dur||j|q|S)zFReturns all the `Node`s whose output this node immediately depends on.N)r3r4rrr1r2)r6Z node_depsr;rrrrr parent_nodesxs zNode.parent_nodesccs8|jD]}|j}|j}|j}|j}||||fVqdS)zYields tuples representing the data inbound from other nodes. Yields: tuples like: (inbound_layer, node_index, tensor_index, tensor). N)r3r4rrr)r6r;Z keras_historyrrrrrriterate_inbounds zNode.iterate_inboundcCsr|jr|jd\}}||fifSt|j}|jD] \}}||||<qt|j|jf|\}}||fS)z;Maps Keras Tensors to computed Tensors using `tensor_dict`.r) r)r.popcopyr'r Zpack_sequence_asr$r%)r6Z tensor_dictr9_Zflat_argumentsr:argskwargsrrr map_argumentss zNode.map_argumentsc s|j|j}|j|\}}tt|jjdd|}||fddt z t j t j dWnty[t t}td|jjddt|d wfd d }t ||}t|sw|jjsw|g}t|}|S) z4Serializes `Node` for Functional API's `get_config`.rNcsrt|dr |j}|j}|jj|}|d}|jj||jgSt|tj r*| St|t j r7t | S|S)z,Serializes a single Tensor passed to `call`.r4r)hasattrr4rrnamegetrr*npZndarraytolistr+r,r get_value)rkhrnode_keynew_node_index) make_node_keynode_conversion_maprr_serialize_keras_tensors    z/Node.serialize.._serialize_keras_tensor)defaultzLayer z- was passed non-JSON-serializable arguments. zArguments had types: z6. They cannot be serialized out when saving the model.cs\t|r!|j}|j}|jj|}|d}|jj||jg}ntd|g}t |S)Nr) r0r4rrrHrIrr rZ ListWrapper)rrMrrNrOdatarRrErPrQrrserialize_first_arg_tensors  z2Node.serialize..serialize_first_arg_tensor)r$r%rZ_split_out_first_argdictzipZ _call_fn_argsupdater r"jsondumpsrZ get_json_type TypeErrortyperHrZ is_nestedZ#_preserve_input_structure_in_configrZconvert_inner_node_data) r6rPrQrDZinputs argumentsZ kwarg_typesrWrUrrVr serializes:        zNode.serializecCs|jr|jgS|jdS)Nr)r!r#r$r>rrr input_tensorss zNode.input_tensorscCs|jr|jgS|jSr )r!r#r>rrroutput_tensorsszNode.output_tensorscCs.ttj|j}t|dkr|js|dS|S)Nrr)r r"r int_shaperar(r!)r6 input_shapesrrrrdszNode.input_shapescCsttj|jSr )r r"rrcrbr>rrr output_shapesszNode.output_shapescCr=r )rr>rrroutbound_layerszNode.outbound_layercCs$|jrgStdd|jd}|S)NcSs|jjSr )r4rrrrrrsz%Node.inbound_layers..r)r!r r"r$)r6inbound_layersrrrrgs  zNode.inbound_layers)NNN)__name__ __module__ __qualname____doc__r<propertyr3r?r@rFr`rarbrdrerfrgrrrrr $s2 @    E     r c@seZdZdZdZdS)r5a}Tracks the Layer call that created a Tensor, for Keras Graph Networks. During construction of Keras Graph Networks, this metadata is added to each Tensor produced as the output of a Layer, starting with an `InputLayer`. This allows Keras to track how each Tensor was produced, and this information is later retraced by the `keras.engine.Network` class to reconstruct the Keras Graph Network. Attributes: layer: The Layer that produced the Tensor. node_index: The specific call to the Layer that produced this Tensor. Layers can be called multiple times in order to share weights. A new node is created every time a Layer is called. tensor_index: The output index for this Tensor. Always zero if the Layer that produced this Tensor only has one output. Nested structures of Tensors are deterministically assigned an index via `nest.flatten`. rN)rhrirjrk __slots__rrrrr5sr5rcCs t|dS)Nr4)rG)r7rrrr0!s r0)rk collectionsrBr[numpyrJZtensorflow.python.frameworkrrr+rZtensorflow.python.kerasrZtensorflow.python.keras.enginerZ*tensorflow.python.keras.saving.saved_modelrZtensorflow.python.keras.utilsrZtensorflow.python.utilr r r namedtupler5r0rrrrs,         e