tensorplay.distributed.tensor.parallel.parallelize_module

tensorplay.distributed.tensor.parallel.parallelize_module(module: Any, device_mesh: Any = None, parallelize_plan: ParallelStyle | dict[str, ParallelStyle] | None = None, *, src_data_rank: int | None = 0) → Any[source]