Searched defs:BackwardPrefetch (Results 1 – 10 of 10) sorted by relevance
| /external/pytorch/test/distributed/fsdp/ |
| D | test_fsdp_backward_prefetch.py | 215 def _test_backward_prefetch(self, backward_prefetch: BackwardPrefetch):
|
| D | test_fsdp_grad_acc.py | 95 backward_prefetch: Optional[BackwardPrefetch],
|
| D | test_fsdp_fine_tune.py | 340 backward_prefetch: BackwardPrefetch,
|
| D | test_fsdp_use_orig_params.py | 123 backward_prefetch: Optional[BackwardPrefetch],
341 backward_prefetch: Optional[BackwardPrefetch],
|
| D | test_fsdp_core.py | 460 backward_prefetch: Optional[BackwardPrefetch],
|
| D | test_wrap.py | 316 backward_prefetch: BackwardPrefetch,
|
| /external/pytorch/torch/distributed/fsdp/ |
| D | api.py | 72 class BackwardPrefetch(Enum): class
|
| D | _init_utils.py | 514 backward_prefetch: BackwardPrefetch,
|
| D | fully_sharded_data_parallel.py | 420 backward_prefetch: Optional[BackwardPrefetch] = BackwardPrefetch.BACKWARD_PRE,
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_fsdp.py | 1293 backward_prefetch: Optional[BackwardPrefetch] = None,
|