Searched refs:_replicate_tensor (Results 1 – 2 of 2) sorted by relevance
555 def _replicate_tensor( member in Replicate
738 local_tensor = placement._replicate_tensor(local_tensor, device_mesh, idx)