Searched defs:ModuleWrapPolicy (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/distributed/fsdp/ | ||
| D | wrap.py | 187 class ModuleWrapPolicy(_Policy): class |
| D | fully_sharded_data_parallel.py | 418 Union[Callable, ModuleWrapPolicy, CustomPolicy] argument |