Searched refs:verify_exhausted (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | cache_dataset_serialization_test.py | 64 outputs = self.gen_outputs(ds_fn, [], 5, verify_exhausted=False)
74 verify_exhausted=False))
85 ds_fn, [5], 8, verify_exhausted=False, save_checkpoint_at_end=False)
94 verify_exhausted=False))
104 outputs = self.gen_outputs(ds_fn, [], 15, verify_exhausted=False)
114 verify_exhausted=False))
125 ds_fn, [15], 18, verify_exhausted=False, save_checkpoint_at_end=False)
132 verify_exhausted=False)
143 ds_fn, [5], 13, verify_exhausted=False, save_checkpoint_at_end=False)
155 verify_exhausted=False)
[all …]
|
| D | snapshot_dataset_serialization_test.py | 55 outputs = self.gen_outputs(ds_fn, [], 50, verify_exhausted=False)
58 self.gen_outputs(ds_fn, [], 50, ckpt_saved=True, verify_exhausted=True))
66 outputs = self.gen_outputs(ds_fn, [], 50, verify_exhausted=False)
71 t = self.gen_outputs(ds_fn, [], 150, ckpt_saved=True, verify_exhausted=True)
81 ds_fn, [10], 20, verify_exhausted=False, save_checkpoint_at_end=False)
86 self.gen_outputs(ds_fn, [], 90, ckpt_saved=True, verify_exhausted=True))
94 outputs = self.gen_outputs(ds_fn, [], 110, verify_exhausted=False)
99 t = self.gen_outputs(ds_fn, [], 90, ckpt_saved=True, verify_exhausted=True)
111 ds_fn, [110], 120, verify_exhausted=False, save_checkpoint_at_end=False)
117 t = self.gen_outputs(ds_fn, [], 90, ckpt_saved=True, verify_exhausted=True)
[all …]
|
| D | dataset_serialization_test_base.py | 98 verify_exhausted=True): argument
114 verify_exhausted=verify_exhausted)
152 verify_exhausted=True,
158 verify_exhausted=True,
167 verify_exhausted=True): argument
186 verify_exhausted=verify_exhausted)
193 verify_exhausted=True): argument
215 verify_exhausted=verify_exhausted)
222 verify_exhausted=False)
237 if verify_exhausted:
[all …]
|
| D | group_by_window_serialization_test.py | 44 lambda: self._build_dataset(components), 12, verify_exhausted=False)
46 lambda: self._build_dataset(components), 12, verify_exhausted=False)
48 lambda: self._build_dataset(components), 12, verify_exhausted=False)
|
| D | group_by_reducer_serialization_test.py | 48 lambda: self._build_dataset(components), 5, verify_exhausted=True)
50 lambda: self._build_dataset(components), 5, verify_exhausted=True)
52 lambda: self._build_dataset(components), 5, verify_exhausted=True)
|
| D | sequence_dataset_serialization_test.py | 112 lambda: self._build_repeat_dataset(-1), 10, verify_exhausted=False)
114 lambda: self._build_repeat_dataset(-1), 20, verify_exhausted=False)
116 lambda: self._build_repeat_dataset(-1), 20, verify_exhausted=False)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | shuffle_and_repeat_test.py | 38 def _gen_outputs(self, ds_fn, num_outputs, verify_exhausted=True): argument
43 if verify_exhausted:
82 lambda: self._build_ds(10, count=None), 100, verify_exhausted=False)
84 lambda: self._build_ds(20, count=None), 100, verify_exhausted=False)
91 lambda: self._build_ds(10, count=-1), 100, verify_exhausted=False)
93 lambda: self._build_ds(20, count=-1), 100, verify_exhausted=False)
130 output = self._gen_outputs(lambda: ds, 2 * 5, verify_exhausted=False)
|