B ӻd+ @sdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZd ZGd d d ZGd ddeddddgZddZdS)zContains the `Node` class.N)ops) tensor_util)backend)base_layer_utils) json_utils)tf_utils)nest_CONSTANT_VALUEc@seZdZdZdddZeddZeddZd 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|dkr gn|}|dkrin|}|dkr,gn|}||_| o@| |_tdd|}tdd|}tdd||_||_||_t|j|jf|_|j ot |jdkot |jd|_ t sx2|jD](}t|t jrtj|ddrt|qWg|_g|_xNt|jD]@\}}t|r|j|tt|}|} |j|| fqW|jj|x.|jD]$} | jj} | dk rl| j|qlWt |jjd} x,tt|D]\}} t|| |d | _qWd d |jD|_ d d t|jD|_!dS) NcSs|S)N)tr r U/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/keras/engine/node.pyDzNode.__init__..cSs|S)Nr )r r r r rErcSs|S)Nr )r r r r rFrrT)Zignore_call_context)layer node_index tensor_indexcSsg|]}tt|qSr )strid).0r r r r osz!Node.__init__..cSsg|]}tt|qSr )rr)rr r r r rps)"ris_inputr map_structureoutputs call_args call_kwargsflatten_flat_argumentslenrZ is_tf_type _single_positional_tensor_passedrZ#executing_eagerly_outside_functions isinstanceTensorrZneeds_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)selfrrrrobjiZelekt_idkt_indexktZ inbound_layerrZtensorr r r __init__2sL         z Node.__init__cCs|jS)zFTensors input to this node that can be traced back to a `keras.Input`.)r#)r,r r r r)rszNode.keras_inputscCs@g}x6|jD],}|jj}|jj}|dk r ||j|q W|S)zFReturns all the `Node`s whose output this node immediately depends on.N)r)r*rrr'r()r,Z node_depsr1rrr r r parent_nodesws zNode.parent_nodesccs:x4|jD]*}|j}|j}|j}|j}||||fVqWdS)zYields tuples representing the data inbound from other nodes. Yields: tuples like: (inbound_layer, node_index, tensor_index, tensor). N)r)r*rrr)r,r1Z keras_historyrrrr r r iterate_inbounds  zNode.iterate_inboundcCsz|jr&|jd\}}||fifSt|j}x"|jD]\}}||||<q:Wt|j|jf|\}}||fSdS)z;Maps Keras Tensors to computed Tensors using `tensor_dict`.rN) r r$popcopyrrZpack_sequence_asrr)r,Z tensor_dictr/_Zflat_argumentsr0argskwargsr r r map_argumentss  zNode.map_argumentsc s|j|j}|j|\}}tt|jjdd|}||fddt yt j t j dWnDtk rt t}td|jjddt|d YnXfd d }t ||}t|s|jjs|g}t|}|S) z4Serializes `Node` for Functional API's `get_config`.rNcsrt|dr@|j}|j}|jj|}|d}|jj||jgSt|tj rT| St|t j rnt | S|S)z,Serializes a single Tensor passed to `call`.r*r)hasattrr*rrnamegetrr!npZndarraytolistrr"r get_value)r khrnode_keynew_node_index) make_node_keynode_conversion_mapr r _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|rB|j}|j}|jj|}|d}|jj||jg}ntd|g}t |S)Nr) r&r*rrr<r=rr rZ ListWrapper)r rArrBrCdata)rFr9rDrEr r serialize_first_arg_tensors z2Node.serialize..serialize_first_arg_tensor)rrrZ_split_out_first_argdictzipZ _call_fn_argsupdaterrjsondumpsrZ get_json_type TypeErrortyper<rZ is_nestedZ#_preserve_input_structure_in_configrZconvert_inner_node_data) r,rDrEr8inputs argumentsZ kwarg_typesrJrIr )rFr9rDrEr serializes(   (   zNode.serializecCs|jr|jgS|jdS)Nr)rrr)r,r r r input_tensorsszNode.input_tensorscCs|jr|jgS|jS)N)rr)r,r r r output_tensorsszNode.output_tensorscCs.ttj|j}t|dkr*|js*|dS|S)Nrr)rrr int_shaperUrr)r, input_shapesr r r rXszNode.input_shapescCsttj|jS)N)rrrrWrV)r,r r r output_shapesszNode.output_shapescCs|jS)N)r)r,r r r outbound_layerszNode.outbound_layercCs$|jr gStdd|jd}|S)NcSs|jjS)N)r*r)r r r r rrz%Node.inbound_layers..r)rrrr)r,inbound_layersr r r r[s   zNode.inbound_layers)NNN)__name__ __module__ __qualname____doc__r2propertyr)r3r4r:rTrUrVrXrYrZr[r r r r r #s  <   E     r c@seZdZdZdZdS)r+a}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`. r N)r\r]r^r_ __slots__r r r r r+sr+rrrcCs t|dS)Nr*)r;)r-r r r r& sr&)r_ collectionsr6rNnumpyr>Ztensorflow.python.frameworkrrZtensorflow.python.kerasrZtensorflow.python.keras.enginerZ*tensorflow.python.keras.saving.saved_modelrZtensorflow.python.keras.utilsrZtensorflow.python.utilrr r namedtupler+r&r r r r s$       e