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