Searched defs:local_shards (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/distributed/tensor/ | ||
| D | _shards_wrapper.py | 209 def local_shards(self) -> List[torch.Tensor]: member in LocalShardsWrapper |
| /external/pytorch/torch/distributed/_shard/sharded_tensor/ | ||
| D | api.py | 127 def local_shards(self) -> List[Shard]: member in ShardedTensorBase |