Home
last modified time | relevance | path

Searched refs:test_checkpoint (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/.ci/pytorch/
H A Dmultigpu-test.sh29 time python test/run_test.py --verbose -i distributed/checkpoint/test_checkpoint
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Dbase_delegate_test.py48 def test_checkpoint(self): member in BaseDelegateTest
/aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/
H A Dsaveable_compat_test.py74 def test_checkpoint(self): member in TestForceCheckpointConversionFlag
/aosp_15_r20/external/pytorch/test/
H A Dtest_functionalization_of_rng_ops.py288 def test_checkpoint(self, dtype, device): member in TestFunctionalizationRngOps
H A Dtest_utils.py195 def test_checkpoint(self): member in TestCheckpoint
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_vmap.py236 def test_checkpoint(self): member in TestVmapAPI