ShardedTensorMetadata (definition) in projects: external

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

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