Searched refs:save_checkpoint_at_end (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | cache_dataset_serialization_test.py | 85 ds_fn, [5], 8, verify_exhausted=False, save_checkpoint_at_end=False)
137 ds_fn, [15], 18, verify_exhausted=False, save_checkpoint_at_end=False)
156 ds_fn, [5], 13, verify_exhausted=False, save_checkpoint_at_end=False)
|
| D | shuffle_dataset_serialization_test.py | 100 save_checkpoint_at_end=False)
|
| D | dataset_serialization_test_base.py | 488 save_checkpoint_at_end=True): argument
550 if save_checkpoint_at_end or i < len(break_points):
|