PrepareModuleOutput

class tensorplay.distributed.tensor.parallel.PrepareModuleOutput(*, output_layouts: Placement | tuple[Placement | None, ...], desired_output_layouts: Placement | tuple[Placement, ...], use_local_output: bool = True)[source]