
    Αi                        S SK Jr  S SKJr  \" SS9 " S S5      5       r\" SS9 " S S5      5       r\ " S	 S
5      5       rg)    )annotations)	dataclassT)frozenc                  b    \ rS rSr% SrS\S'   S\S'   S\S'   SrS	\S
'   SrS\S'   SrS	\S'   Sr	g)LocalTensorMetadata   z6
The location of a local tensor in the global tensor.

tuple[int]global_offsetlocal_shapestrdtypeNtuple[int] | Noneglobal_shapeFboolis_flattenedflattened_range )
__name__
__module____qualname____firstlineno____doc____annotations__r   r   r   __static_attributes__r       o/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/distributed/flex_checkpoint/dcp/metadata.pyr   r      s:     J&*L#*L$)-O&-r   r   c                  f    \ rS rSr% SrS\S'   S\S'   SrS\S	'   S
rS\S'   S
rS\S'   S
r	S\S'   Sr
g
)LocalTensorIndex!   z#
The identifier of a local tensor.
r   
tensor_keyr	   r
   Fr   r   Nr   r   z
int | None
replica_idr   r   )r   r   r   r   r   r   r   r   r!   r   r   r   r   r   r   r   !   s>     OL$)-O&-!J
!%)K")r   r   c                  @    \ rS rSr% SrS\S'   SrS\S'   SrS\S'   S	rg)
Metadata/   Nz$dict[str, list[LocalTensorMetadata]]state_dict_metadatazdict[LocalTensorIndex, str]storage_metadatazdict[str, tuple[str]]flat_mappingr   )	r   r   r   r   r%   r   r&   r'   r   r   r   r   r#   r#   /   s%    @D=D4818*.L'.r   r#   N)
__future__r   dataclassesr   r   r   r#   r   r   r   <module>r*      sb    # ! $
. 
. 
. $
* 
* 
* / / /r   