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 | 63 outputs = self.gen_outputs(ds_fn, [], 5, verify_exhausted=False)
73 verify_exhausted=False))
85 ds_fn, [5], 8, verify_exhausted=False, save_checkpoint_at_end=False)
95 verify_exhausted=False))
105 verify_exhausted=False)
115 outputs = self.gen_outputs(ds_fn, [], 15, verify_exhausted=False)
125 verify_exhausted=False))
137 ds_fn, [15], 18, verify_exhausted=False, save_checkpoint_at_end=False)
144 verify_exhausted=False)
156 ds_fn, [5], 13, verify_exhausted=False, save_checkpoint_at_end=False)
[all …]
|
| D | dataset_serialization_test_base.py | 114 verify_exhausted=True): argument
130 verify_exhausted=verify_exhausted)
168 verify_exhausted=True,
174 verify_exhausted=True,
182 verify_exhausted=True): argument
200 verify_exhausted=verify_exhausted)
207 verify_exhausted=True): argument
226 verify_exhausted=verify_exhausted)
233 verify_exhausted=True): argument
255 verify_exhausted=verify_exhausted)
[all …]
|
| D | group_by_window_serialization_test.py | 39 lambda: self._build_dataset(components), 12, verify_exhausted=False)
41 lambda: self._build_dataset(components), 12, verify_exhausted=False)
43 lambda: self._build_dataset(components), 12, verify_exhausted=False)
45 lambda: self._build_dataset(components), 12, verify_exhausted=False)
47 lambda: self._build_dataset(components), 12, verify_exhausted=False)
53 verify_exhausted=False)
|
| D | group_by_reducer_serialization_test.py | 43 lambda: self._build_dataset(components), 5, verify_exhausted=True)
45 lambda: self._build_dataset(components), 5, verify_exhausted=True)
47 lambda: self._build_dataset(components), 5, verify_exhausted=True)
49 lambda: self._build_dataset(components), 5, verify_exhausted=True)
51 lambda: self._build_dataset(components), 5, verify_exhausted=True)
57 verify_exhausted=True)
|
| D | sequence_dataset_serialization_test.py | 106 lambda: self._build_repeat_dataset(-1), 10, verify_exhausted=False)
108 lambda: self._build_repeat_dataset(-1), 10, verify_exhausted=False)
110 lambda: self._build_repeat_dataset(-1), 20, verify_exhausted=False)
112 lambda: self._build_repeat_dataset(-1), 20, verify_exhausted=False)
117 verify_exhausted=False)
|
| D | shuffle_dataset_serialization_test.py | 99 verify_exhausted=False,
106 verify_exhausted=False)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | shuffle_and_repeat_test.py | 37 def _gen_outputs(self, ds_fn, num_outputs, verify_exhausted=True): argument
42 if verify_exhausted:
76 lambda: self._build_ds(10, count=None), 100, verify_exhausted=False)
78 lambda: self._build_ds(20, count=None), 100, verify_exhausted=False)
84 lambda: self._build_ds(10, count=-1), 100, verify_exhausted=False)
86 lambda: self._build_ds(20, count=-1), 100, verify_exhausted=False)
119 output = self._gen_outputs(lambda: ds, 2 * 5, verify_exhausted=False)
|