Searched defs:__get_tensor_shard__ (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/distributed/ | ||
| D | _checkpointable.py | 32 def __get_tensor_shard__(self, index) -> torch.Tensor: member in _Checkpointable |
| /external/pytorch/torch/distributed/tensor/ | ||
| D | _shards_wrapper.py | 278 def __get_tensor_shard__(self, index: MetadataIndex) -> torch.Tensor: member in LocalShardsWrapper |
| D | _api.py | 614 def __get_tensor_shard__(self, index): member in DTensor |