Searched defs:ds_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | checkpoint_test_base.py | 61 def disable_optimizations(ds_fn): argument
70 def verify_unused_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
80 def verify_fully_used_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
90 def verify_exhausted_iterator(obj, ds_fn, num_outputs, sparse_tensors=False): argument
100 def verify_multiple_breaks(obj, ds_fn, num_outputs, sparse_tensors=False): argument
111 ds_fn, argument
141 ds_fn, argument
164 ds_fn, argument
185 def verify_exhausted_iterator(self, ds_fn, num_outputs, sparse_tensors=False): argument
212 ds_fn, argument
[all …]
|
| D | snapshot_test.py | 1021 def ds_fn(): function
1115 def ds_fn(): function
|
| D | shuffle_test.py | 465 def ds_fn(): function
|
| D | cache_test.py | 452 def ds_fn(): function
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | shuffle_and_repeat_test.py | 39 def _gen_outputs(self, ds_fn, num_outputs, verify_exhausted=True): argument
|