Searched defs:FSDPExtensions (Results 1 – 2 of 2) sorted by relevance
16 class FSDPExtensions(ABC): class91 def _set_fsdp_extensions(flattener: FSDPExtensions) -> None:98 fsdp_extension: Optional[FSDPExtensions] = None,110 fsdp_extension: Optional[FSDPExtensions] = None,123 fsdp_extension: Optional[FSDPExtensions] = None,143 fsdp_extension: Optional[FSDPExtensions] = None,159 fsdp_extension: Optional[FSDPExtensions] = None,172 fsdp_extension: Optional[FSDPExtensions] = None,
497 fsdp_extension: Optional[FSDPExtensions] = None,