Searched defs:ShardingPlan (Results 1 – 2 of 2) sorted by relevance
11 class ShardingPlan: class
226 def shard_module(module: nn.Module, plan: ShardingPlan, src_rank=0, process_group=None):