Searched defs:__get_tensor_shard__ (Results 1 – 3 of 3) sorted by relevance
32 def __get_tensor_shard__(self, index) -> torch.Tensor: member in _Checkpointable
278 def __get_tensor_shard__(self, index: MetadataIndex) -> torch.Tensor: member in LocalShardsWrapper
614 def __get_tensor_shard__(self, index): member in DTensor