build_plan (definition) in projects: aosp_15_r20

Searched defs:build_plan (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_plan/
H A D	api.py	`77 def build_plan(self, module: nn.Module) -> ShardingPlan: member in ShardingPlanner`
/aosp_15_r20/external/pytorch/test/distributed/_shard/sharding_plan/
H A D	test_sharding_plan.py	`42 def build_plan(self, module: nn.Module) -> ShardingPlan: member in ChunkAllShardingPlanner`
/aosp_15_r20/external/federated-compute/fcp/artifact_building/
H A D	federated_compute_plan_builder.py	`1475 def build_plan( function`