o
    bi                     @   s   d dl ZG dd dZdS )    Nc                   @   s   e Zd Zdd ZdddZdS )TFExportArchivec                 C   sF   |j }|j}|j}| j j |7  _ | j j|7  _| j j|7  _d S N)	variablestrainable_variablesnon_trainable_variables_tf_trackable)selflayerr   r   r    r
   W/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/backend/tensorflow/export.py_track_layer   s   zTFExportArchive._track_layerNc                 K   s   t j||dd}|S )NF)input_signature	autograph)tffunction)r   namefnr   kwargsdecorated_fnr
   r
   r   add_endpoint   s   zTFExportArchive.add_endpointr   )__name__
__module____qualname__r   r   r
   r
   r
   r   r      s    
r   )
tensorflowr   r   r
   r
   r
   r   <module>   s    