Home
last modified time | relevance | path

Searched defs:OffloadPolicy (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D_fsdp_api.py59 class OffloadPolicy: class
H A Dfully_shard.py38 offload_policy: OffloadPolicy = OffloadPolicy(),
H A D_fsdp_param.py220 offload_policy: OffloadPolicy,
H A D_fsdp_param_group.py110 offload_policy: OffloadPolicy,
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_training.py322 offload_policy: OffloadPolicy,
679 offload_policy: OffloadPolicy,
/aosp_15_r20/external/pytorch/test/distributed/_tools/
H A Dtest_fsdp2_mem_tracker.py76 offload_policy: OffloadPolicy,