U JcU @sxddlmZmZddlmZddlmZddlZddlm Z GdddeZ eGdd d e Z eGd d d e Z dS) ) dataclassfield)Enum)ListN) ShardMetadatac@seZdZdZdZdZdS)MEM_FORMAT_ENCODINGrN)__name__ __module__ __qualname__TORCH_CONTIGUOUS_FORMATTORCH_CHANNELS_LASTTORCH_PRESERVE_FORMATrrT/tmp/pip-unpacked-wheel-gikjz4vx/torch/distributed/_shard/sharded_tensor/metadata.pyrsrc@seZdZUdZeedZejed<eej dZ ej ed<dZ e ed<eej dZejed<dZe ed<d d Zd d Zeejdd ddZdS)TensorPropertiesz+ Properties used to create :class:`Tensor` defaultdtypelayoutF requires_grad memory_format pin_memorycCs`|j}|tjkrtj}n2|tjkr*tj}n |tjkrs   8