o
    "j                     @   s`   d dl mZ d dlmZmZmZ eG dd dZeddG dd dZeG d	d
 d
ZdS )    )	dataclass)DictListTuplec                   @   s*   e Zd ZU dZee ed< ee ed< dS )LocalTensorMetadataz>
    The location of a local tensor in the global tensor.
    global_offsetZlocal_shapeN)__name__
__module____qualname____doc__r   int__annotations__ r   r   g/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/distributed/checkpoint/metadata.pyr      s   
 r   T)frozenc                   @   s&   e Zd ZU dZeed< ee ed< dS )LocalTensorIndexz+
    The identifier of a local tensor.
    Z
tensor_keyr   N)r   r	   r
   r   strr   r   r   r   r   r   r   r      s   
 r   c                   @   s:   e Zd ZU dZeeee f ed< dZ	ee
ef ed< dS )MetadataNstate_dict_metadatastorage_metadata)r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   '   s   
 r   N)	dataclassesr   typingr   r   r   r   r   r   r   r   r   r   <module>   s   		