Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_fsdp.py67 class FSDPInitMode(Enum): class
305 fsdp_init_mode: FSDPInitMode,
434 fsdp_init_mode: FSDPInitMode,
482 fsdp_init_mode: FSDPInitMode,
564 fsdp_init_mode: FSDPInitMode,
682 fsdp_init_mode: FSDPInitMode,
798 fsdp_init_mode: FSDPInitMode,
1287 fsdp_init_mode: FSDPInitMode,