ShardedTensorMetadata (definition) in projects: aosp_15_r20

Searched defs:ShardedTensorMetadata (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/
H A D	api.py	`139 sharded_tensor_metadata: ShardedTensorMetadata, 880 sharded_tensor_metadata: ShardedTensorMetadata,`
H A D	metadata.py	`84 class ShardedTensorMetadata: class`
H A D	utils.py	`203 gathered_metadatas: Sequence[Optional[ShardedTensorMetadata]],`