| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | from_generator_test.py | 39 requires_initialization): argument
45 requires_initialization=requires_initialization,
52 num_repeats=[1, 2], requires_initialization=[True, False])))
53 def testFromGeneratorUsingFn(self, num_repeats, requires_initialization): argument
64 requires_initialization=requires_initialization)
70 num_repeats=[1, 2], requires_initialization=[True, False])))
71 def testFromGeneratorUsingList(self, num_repeats, requires_initialization): argument
78 requires_initialization=requires_initialization)
84 num_repeats=[1, 2], requires_initialization=[True, False])))
85 def testFromGeneratorUsingNdarray(self, num_repeats, requires_initialization): argument
[all …]
|
| D | test_base.py | 92 def getNext(self, dataset, requires_initialization=False, shared_name=None): argument
130 if requires_initialization:
151 def getDatasetOutput(self, dataset, requires_initialization=False): argument
153 dataset, requires_initialization=requires_initialization)
171 requires_initialization=False, argument
210 dataset, requires_initialization=requires_initialization)
220 dataset, requires_initialization=requires_initialization)
|
| D | list_files_test.py | 56 self.getNext(dataset, requires_initialization=True)
102 next_element = self.getNext(dataset_fn(), requires_initialization=True)
128 requires_initialization=True)
|
| D | map_test.py | 387 get_next = self.getNext(ds, requires_initialization=True)
411 get_next = self.getNext(dataset, requires_initialization=True)
429 get_next = self.getNext(dataset, requires_initialization=True)
455 get_next = self.getNext(dataset, requires_initialization=True)
468 get_next = self.getNext(dataset, requires_initialization=True)
476 get_next = self.getNext(dataset, requires_initialization=True)
515 get_next = self.getNext(dataset, requires_initialization=True)
1076 get_next = self.getNext(dataset, requires_initialization=True)
1122 get_next = self.getNext(dataset, requires_initialization=True)
1156 get_next = self.getNext(dataset, requires_initialization=True)
[all …]
|
| D | cache_test.py | 150 get_next1 = self.getNext(cache_dataset1, requires_initialization=True)
151 get_next2 = self.getNext(cache_dataset2, requires_initialization=True)
228 cached_next = self.getNext(cached_dataset, requires_initialization=True)
230 uncached_dataset, requires_initialization=True)
308 get_next = self.getNext(dataset, requires_initialization=True)
|
| D | scan_test.py | 129 requires_initialization=True,
161 requires_initialization=True,
|
| D | dataset_test.py | 83 round_tripped, [10, 11, -1], requires_initialization=True)
344 requires_initialization=True)
348 requires_initialization=True)
|
| D | from_tensors_test.py | 80 dataset, expected_output=[[1.0, 2.0]], requires_initialization=True)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | make_deterministic_test.py | 89 requires_initialization=True)
115 requires_initialization=True)
128 get_next = self.getNext(dataset, requires_initialization=True)
155 requires_initialization=True)
184 requires_initialization=True)
221 requires_initialization=True)
358 get_next = self.getNext(dataset, requires_initialization=True)
384 dataset, expected_output=expected_output, requires_initialization=True)
|
| D | map_parallelization_test.py | 100 requires_initialization=True)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| D | coordinated_read_ft_test.py | 38 get_next = self.getNext(ds, requires_initialization=True)
69 get_next = self.getNext(ds, requires_initialization=True)
101 get_next = self.getNext(ds, requires_initialization=True)
|
| D | coordinated_read_test.py | 40 get_next = self.getNext(ds, requires_initialization=False)
51 get_next = self.getNext(ds, requires_initialization=False)
57 get_next_ds2 = self.getNext(ds2, requires_initialization=False)
|
| D | multi_device_test.py | 71 dataset, [10, 11, -1], requires_initialization=True)
80 self.getDatasetOutput(dataset, requires_initialization=True)
|
| D | data_service_ops_test.py | 136 self.assertDatasetProduces(ds, [10, 11, -1], requires_initialization=True)
163 ds, [v1, v2, default_value], requires_initialization=True)
1315 ds, list(range(10, 13)), requires_initialization=True)
|
| D | dynamic_sharding_test.py | 415 requires_initialization=True,
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | sleep_test.py | 51 get_next = self.getNext(ds, requires_initialization=True)
71 get_next = self.getNext(ds, requires_initialization=True)
|
| D | replicate_test.py | 121 dataset1, range(1, 101), requires_initialization=True)
124 dataset2, range(1, 101), requires_initialization=True)
129 dataset0, range(1, 101), requires_initialization=True)
322 dataset0, range(1, 101), requires_initialization=True)
325 dataset1, range(1, 101), requires_initialization=True)
328 dataset2, range(1, 101), requires_initialization=True)
|
| D | wrap_unwrap_test.py | 45 get_next = self.getNext(variant_ds, requires_initialization=True)
|
| D | map_and_batch_test.py | 234 get_next = self.getNext(dataset, requires_initialization=True)
374 get_next = self.getNext(dataset, requires_initialization=True)
|
| D | auto_shard_dataset_test.py | 601 self.getDatasetOutput(dataset, requires_initialization=True)
|