Searched refs:implicit_replication (Results 1 – 6 of 6) sorted by relevance
14 def implicit_replication(): function30 implicit_replication.__module__ = "torch.distributed.tensor.experimental"
11 from torch.distributed._tensor.experimental import implicit_replication174 with implicit_replication():
35 from torch.distributed._tensor.experimental import implicit_replication1095 with implicit_replication():
11 from torch.distributed._tensor.experimental import implicit_replication464 implicit_replication()
262 from torch.distributed._tensor.experimental import implicit_replication264 with implicit_replication():
784 from torch.distributed._tensor.experimental import implicit_replication786 with implicit_replication():