Searched refs:test_checkpoint (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/.ci/pytorch/ | ||
| H A D | multigpu-test.sh | 29 time python test/run_test.py --verbose -i distributed/checkpoint/test_checkpoint |
| /aosp_15_r20/external/tensorflow/tensorflow/python/trackable/ | ||
| H A D | base_delegate_test.py | 48 def test_checkpoint(self): member in BaseDelegateTest |
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ | ||
| H A D | saveable_compat_test.py | 74 def test_checkpoint(self): member in TestForceCheckpointConversionFlag |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_functionalization_of_rng_ops.py | 288 def test_checkpoint(self, dtype, device): member in TestFunctionalizationRngOps |
| H A D | test_utils.py | 195 def test_checkpoint(self): member in TestCheckpoint |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_vmap.py | 236 def test_checkpoint(self): member in TestVmapAPI |