Searched defs:test_checkpoint (Results 1 – 5 of 5) sorted by relevance
| /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 |