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())
|