Searched defs:build_plan (Results 1 – 3 of 3) sorted by relevance
77 def build_plan(self, module: nn.Module) -> ShardingPlan: member in ShardingPlanner
42 def build_plan(self, module: nn.Module) -> ShardingPlan: member in ChunkAllShardingPlanner
1475 def build_plan( function