deepfold.distributed.model_parallel.broadcast_on_model_parallel_region

deepfold.distributed.model_parallel.broadcast_on_model_parallel_region(tensor: Tensor, src_rank: int) Tensor[source]