Searched defs:use_reentrant (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 395 self, model, input_var, target, loss, run_checkpoint=False, use_reentrant=True argument
413 use_reentrant=True, argument
470 def __init__(self, use_reentrant=True): argument
488 def __init__(self, use_reentrant=True): argument
502 def __init__(self, use_reentrant=True): argument
514 def __init__(self, use_reentrant=True): argument
528 def __init__(self, use_reentrant=True): argument
545 def test_ddp_checkpointing_once(self, use_reentrant): argument
570 def test_ddp_checkpointing_unused_params(self, use_reentrant): argument
603 def test_ddp_checkpointing_twice(self, use_reentrant): argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_checkpoint_wrapper.py | 181 def test(use_checkpointing, use_wrapper, use_reentrant): argument
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | checkpoint.py | 504 def checkpoint_sequential(functions, segments, input, use_reentrant=None, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_utils.py | 81 use_reentrant, argument
|
| H A D | test_autograd.py | 6790 def __init__(self, n, use_checkpoint, use_reentrant): argument
7143 def test_checkpointing_without_reentrant_detached_tensor(self, use_reentrant): argument
|