tensorplay.distributed.pipelining.pipeline

tensorplay.distributed.pipelining.pipeline(module: Any, mb_args: tuple[Any, ...], mb_kwargs: dict[str, Any] | None = None, split_spec: dict[str, SplitPoint] | None = None, split_policy: Any = None) → Pipe[source]