Searched defs:local_tensor (Results 1 – 4 of 4) sorted by relevance
204 local_tensor, check_shape_stride=False argument
435 def assert_local_shard_across_ranks(local_tensor, group, check_equal=True): argument
520 Tensor local_tensor; in LockedWriteOrAggregate() local
1123 def local_tensor(self) -> torch.Tensor: member in ShardedTensor