Searched refs:set_modules_to_backward_prefetch (Results 1 – 2 of 2) sorted by relevance
307 def set_modules_to_backward_prefetch(self, modules: List["FSDPModule"]) -> None: member in FSDPModule
736 layer.set_modules_to_backward_prefetch(layers_to_prefetch)