tensorplay.distributed.pipelining.build_stage

tensorplay.distributed.pipelining.build_stage(stage_module: Any, stage_index: int, pipe_info: Any, device: Any = None, group: Any = None) → _PipelineStage[source]