Searched refs:ds_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | cache_dataset_serialization_test.py | 46 def ds_fn(): function
50 return ds_fn
60 ds_fn = self.make_dataset_fn(is_memory)
63 outputs = self.gen_outputs(ds_fn, [], 5, verify_exhausted=False)
70 ds_fn, [],
81 ds_fn = self.make_dataset_fn(is_memory)
85 ds_fn, [5], 8, verify_exhausted=False, save_checkpoint_at_end=False)
92 ds_fn, [],
102 ds_fn, [],
112 ds_fn = self.make_dataset_fn(is_memory)
[all …]
|
| D | dataset_serialization_test_base.py | 111 ds_fn, argument
127 ds_fn, [0],
132 def verify_fully_used_iterator(self, ds_fn, num_outputs, argument
150 ds_fn, [num_outputs], num_outputs, sparse_tensors=sparse_tensors)
152 def verify_exhausted_iterator(self, ds_fn, num_outputs, sparse_tensors=False): argument
166 ds_fn, [],
171 ds_fn, [],
179 ds_fn, argument
195 ds_fn,
203 ds_fn, argument
[all …]
|
| D | shuffle_dataset_serialization_test.py | 81 def ds_fn(): function
95 ds_fn,
102 ds_fn,
118 def ds_fn(): function
129 ds = ds_fn()
|
| D | parallel_map_dataset_serialization_test.py | 65 for ds_fn in [self._build_ds, self._build_ds_with_prefetch]:
67 ds_fn,
68 lambda: ds_fn(multiplier=15.0), # pylint: disable=cell-var-from-loop
|
| /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
38 get_next = self.getNext(ds_fn())
|