Home
last modified time | relevance | path

Searched refs:assert_items_equal (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/
Ddynamic_sharding_test.py50 ds, list(range(num_elements)), assert_items_equal=True)
58 self.assertDatasetProduces(ds, vals, assert_items_equal=True)
67 self.assertDatasetProduces(ds, elements, assert_items_equal=True)
78 self.assertDatasetProduces(ds, elements, assert_items_equal=True)
87 self.assertDatasetProduces(ds, elements, assert_items_equal=True)
114 ds, num_repeats * list(range(num_elements)), assert_items_equal=True)
125 ds, num_repeats * list(range(num_elements)), assert_items_equal=True)
172 ds, num_repeats * list(range(num_elements)), assert_items_equal=True)
287 assert_items_equal = (num_workers > 1)
289 ds, expected, assert_items_equal=assert_items_equal)
[all …]
Dlocal_workers_test.py55 assert_items_equal=True)
75 assert_items_equal=True)
93 assert_items_equal=True)
111 assert_items_equal=True)
129 ds, list(range(num_elements)), assert_items_equal=True)
184 ds, num_workers * list(range(num_elements)), assert_items_equal=True)
235 assert_items_equal=True)
239 assert_items_equal=True)
243 assert_items_equal=True)
Dworker_tags_test.py69 assert_items_equal=True)
105 dataset, list(range(num_elements)), assert_items_equal=True)
194 dataset, 4 * list(range(num_elements)), assert_items_equal=True)
Dmulti_process_cluster_test.py46 assert_items_equal=True)
Dauto_shard_test.py189 dataset, list(range(20)), assert_items_equal=True)
234 self.assertDatasetProduces(dataset, expected, assert_items_equal=True)
324 self.assertDatasetProduces(dataset, expected, assert_items_equal=True)
376 self.assertDatasetProduces(dataset, expected, assert_items_equal=True)
Ddata_service_ops_test.py259 ds, num_workers * list(range(num_elements)), assert_items_equal=True)
274 ds, num_workers * list(range(num_elements)), assert_items_equal=True)
819 assert_items_equal=True)
896 self.assertDatasetProduces(ds, list(range(40)), assert_items_equal=True)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dlist_files_test.py70 assert_items_equal=True)
142 assert_items_equal=True)
156 assert_items_equal=True)
170 assert_items_equal=True)
215 assert_items_equal=True)
230 assert_items_equal=True)
Dtest_base.py139 assert_items_equal): argument
140 if assert_items_equal:
173 assert_items_equal=False, argument
230 self._compareOutputToExpected(result, expected_output, assert_items_equal)
Dtext_line_dataset_test.py134 dataset, expected_output=expected_output * 10, assert_items_equal=True)
182 ds, expected_output=expected_output, assert_items_equal=True)
190 ds, expected_output=expected_output, assert_items_equal=True)
Dtf_record_dataset_test.py170 dataset, expected_output=expected_output * 10, assert_items_equal=True)
179 ds, expected_output=expected_output, assert_items_equal=True)
188 ds, expected_output=expected_output, assert_items_equal=True)
Dfixed_length_record_dataset_test.py175 assert_items_equal=True)
211 assert_items_equal=True)
Dshuffle_test.py253 self.assertDatasetProduces(dataset, [0, 0, 1], assert_items_equal=True)
266 self.assertDatasetProduces(dataset, [0, 0, 1], assert_items_equal=True)
Dsnapshot_test.py753 self.assertDatasetProduces(dataset3, expected, assert_items_equal=True)
825 self.assertDatasetProduces(dataset, expected, assert_items_equal=True)
839 self.assertDatasetProduces(dataset2, expected, assert_items_equal=True)
880 self.assertDatasetProduces(dataset2, expected, assert_items_equal=True)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmatching_files_dataset_test.py80 assert_items_equal=True)
97 assert_items_equal=True)
114 assert_items_equal=True)