| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | shard_test.py | 34 self.assertDatasetProduces(dataset, expected_output=[2, 7])
41 self.assertDatasetProduces(dataset, expected_output=[(2, 8), (7, 3)])
46 self.assertDatasetProduces(dataset, expected_output=[0, 5])
75 self.assertDatasetProduces(dataset, expected_output=[])
80 self.assertDatasetProduces(dataset, expected_output=[5])
85 self.assertDatasetProduces(dataset, expected_output=[4, 9])
90 self.assertDatasetProduces(dataset, expected_output=[3, 7])
95 self.assertDatasetProduces(dataset, expected_output=[5])
|
| D | tf_record_dataset_test.py | 87 self.assertDatasetProduces(
93 self.assertDatasetProduces(
103 self.assertDatasetProduces(dataset, expected_output=expected_output)
112 self.assertDatasetProduces(dataset, expected_output=expected_output * 10)
122 self.assertDatasetProduces(dataset, expected_output=expected_output * 10)
140 self.assertDatasetProduces(dataset, expected_output=expected_output)
156 self.assertDatasetProduces(dataset, expected_output=expected_output)
167 self.assertDatasetProduces(dataset, expected_output=expected_output)
177 self.assertDatasetProduces(dataset, expected_output=expected_output)
188 self.assertDatasetProduces(
[all …]
|
| D | unbatch_test.py | 43 self.assertDatasetProduces(dataset, range(4))
55 self.assertDatasetProduces(data, [(i,) * 3 for i in range(10)])
62 self.assertDatasetProduces(data, list(range(10)) * 10)
75 self.assertDatasetProduces(
91 self.assertDatasetProduces(data, expected_output=expected_output)
108 self.assertDatasetProduces(
124 self.assertDatasetProduces(
137 self.assertDatasetProduces(data, [((i,),) * 3 for i in range(10)])
152 self.assertDatasetProduces(
162 self.assertDatasetProduces(data, [])
[all …]
|
| D | text_line_dataset_test.py | 97 self.assertDatasetProduces(
101 self.assertDatasetProduces(
108 self.assertDatasetProduces(
114 self.assertDatasetProduces(
118 self.assertDatasetProduces(
131 self.assertDatasetProduces(
142 self.assertDatasetProduces(repeat_dataset, expected_output=expected_output)
179 self.assertDatasetProduces(
|
| D | batch_test.py | 117 self.assertDatasetProduces(dataset, expected_output=range(10))
132 self.assertDatasetProduces(dataset, expected_output=expected_output)
158 self.assertDatasetProduces(dataset, expected_output=expected_output)
175 self.assertDatasetProduces(dataset, expected_output=expected_output)
188 self.assertDatasetProduces(
206 self.assertDatasetProduces(dataset, expected_output=expected_output)
216 self.assertDatasetProduces(dataset, expected_output=expected_output)
229 self.assertDatasetProduces(dataset, expected_output=expected_output)
235 self.assertDatasetProduces(dataset, expected_output=[list(range(10))])
|
| D | list_files_test.py | 68 self.assertDatasetProduces(
83 self.assertDatasetProduces(
128 self.assertDatasetProduces(
140 self.assertDatasetProduces(
154 self.assertDatasetProduces(
168 self.assertDatasetProduces(
213 self.assertDatasetProduces(
228 self.assertDatasetProduces(
|
| D | range_test.py | 43 self.assertDatasetProduces(dataset, expected_output=expected_output)
56 self.assertDatasetProduces(dataset, expected_output=expected_output)
71 self.assertDatasetProduces(dataset, expected_output=expected_output)
99 self.assertDatasetProduces(dataset, expected_output=expected_output)
112 self.assertDatasetProduces(dataset, expected_output=expected_output)
127 self.assertDatasetProduces(dataset, expected_output=expected_output)
142 self.assertDatasetProduces(dataset, expected_output=expected_output)
|
| D | fixed_length_record_dataset_test.py | 92 self.assertDatasetProduces(
99 self.assertDatasetProduces(
110 self.assertDatasetProduces(
157 self.assertDatasetProduces(dataset, expected_output=expected_output)
173 self.assertDatasetProduces(dataset, expected_output=expected_output,
185 self.assertDatasetProduces(
209 self.assertDatasetProduces(dataset, expected_output=expected_output,
|
| D | window_test.py | 141 self.assertDatasetProduces(dataset, expected_output=expected_output)
171 self.assertDatasetProduces(dataset, expected_output=expected_output)
200 self.assertDatasetProduces(dataset, expected_output=expected_output)
213 self.assertDatasetProduces(
227 self.assertDatasetProduces(
239 self.assertDatasetProduces(x, range(i*10, (i+1)*10))
240 self.assertDatasetProduces(y, range(i*10, (i+1)*10))
249 self.assertDatasetProduces(
|
| D | from_tensors_test.py | 54 self.assertDatasetProduces(dataset, expected_output=[components])
61 self.assertDatasetProduces(dataset, expected_output=range(10))
72 self.assertDatasetProduces(
92 self.assertDatasetProduces(dataset, expected_output=[components])
113 self.assertDatasetProduces(dataset, expected_output=[components])
124 self.assertDatasetProduces(dataset, expected_output=[components])
141 self.assertDatasetProduces(dataset, expected_output=[components])
273 self.assertDatasetProduces(dataset, range(100))
|
| D | filter_test.py | 98 self.assertDatasetProduces(dataset, expected_output=[0, 1, 3])
106 self.assertDatasetProduces(
123 self.assertDatasetProduces(dataset, expected_output=[input_data[0]])
140 self.assertDatasetProduces(
149 self.assertDatasetProduces(
|
| D | flat_map_test.py | 53 self.assertDatasetProduces(dataset, expected_output=expected_output)
67 self.assertDatasetProduces(dataset, expected_output=expected_output)
130 self.assertDatasetProduces(dataset, expected_output=expected_output)
152 self.assertDatasetProduces(dataset, expected_output=expected_output)
169 self.assertDatasetProduces(dataset, expected_output=expected_output)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | snapshot_test.py | 124 self.assertDatasetProduces(dataset, expected)
134 self.assertDatasetProduces(dataset2, expected)
149 self.assertDatasetProduces(dataset, expected)
155 self.assertDatasetProduces(dataset2, expected)
170 self.assertDatasetProduces(dataset, expected)
181 self.assertDatasetProduces(dataset2, expected)
202 self.assertDatasetProduces(dataset, expected)
244 self.assertDatasetProduces(dataset, [])
253 self.assertDatasetProduces(dataset2, [])
259 self.assertDatasetProduces(dataset, list(range(1000)))
[all …]
|
| D | replicate_test.py | 69 self.assertDatasetProduces(dataset0, range(100))
71 self.assertDatasetProduces(dataset1, range(100))
73 self.assertDatasetProduces(dataset2, range(100))
87 self.assertDatasetProduces(dataset0, range(100))
89 self.assertDatasetProduces(dataset1, range(100))
91 self.assertDatasetProduces(dataset2, range(100))
105 self.assertDatasetProduces(dataset0, range(100))
107 self.assertDatasetProduces(dataset1, range(100))
109 self.assertDatasetProduces(dataset2, range(100))
124 self.assertDatasetProduces(
[all …]
|
| D | rebatch_dataset_test.py | 235 self.assertDatasetProduces(rebatched_dataset, expected_output)
247 self.assertDatasetProduces(rebatched_dataset, expected_output)
259 self.assertDatasetProduces(rebatched_dataset, expected_output)
271 self.assertDatasetProduces(rebatched_dataset, expected_output)
288 self.assertDatasetProduces(rebatched_dataset, expected_output)
307 self.assertDatasetProduces(rebatched_dataset, expected_output)
322 self.assertDatasetProduces(rebatched_dataset, expected_output)
337 self.assertDatasetProduces(rebatched_dataset, expected_output)
351 self.assertDatasetProduces(rebatched_dataset, expected_output)
364 self.assertDatasetProduces(rebatched_dataset, expected_output)
[all …]
|
| D | auto_shard_dataset_test.py | 86 self.assertDatasetProduces(dataset, list(chunk(expected, batch)))
141 self.assertDatasetProduces(dataset, expected)
163 self.assertDatasetProduces(dataset, expected)
221 self.assertDatasetProduces(dataset, expected)
307 self.assertDatasetProduces(dataset, [0, 2])
361 self.assertDatasetProduces(dataset, expected)
379 self.assertDatasetProduces(dataset, expected)
403 self.assertDatasetProduces(dataset, [])
417 self.assertDatasetProduces(dataset, expected)
434 self.assertDatasetProduces(dataset, list(chunk(expected, 5)))
[all …]
|
| D | assert_next_test.py | 39 self.assertDatasetProduces(dataset, expected_output=[0])
50 self.assertDatasetProduces(dataset, expected_output=[[0]])
59 self.assertDatasetProduces(
72 self.assertDatasetProduces(
|
| D | matching_files_test.py | 56 self.assertDatasetProduces(
65 self.assertDatasetProduces(
77 self.assertDatasetProduces(
94 self.assertDatasetProduces(
111 self.assertDatasetProduces(
|
| D | prefetch_with_slack_test.py | 75 self.assertDatasetProduces(dataset, range(10))
86 self.assertDatasetProduces(dataset, range(1, 11))
95 self.assertDatasetProduces(dataset, range(10))
106 self.assertDatasetProduces(dataset, range(10))
|
| D | io_test.py | 54 self.assertDatasetProduces(dataset2, range(42))
71 self.assertDatasetProduces(dataset2, expected)
84 self.assertDatasetProduces(dataset2, expected)
100 self.assertDatasetProduces(dataset, range(42))
|
| D | take_while_test.py | 53 self.assertDatasetProduces(dataset, np.arange(expected_num_elements))
74 self.assertDatasetProduces(dataset, np.arange(upper_bound))
77 self.assertDatasetProduces(dataset, np.arange(upper_bound))
126 self.assertDatasetProduces(dataset, np.tile([0, 1], 5))
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | choose_fastest_branch_dataset_test.py | 47 self.assertDatasetProduces(
68 self.assertDatasetProduces(
84 self.assertDatasetProduces(
102 self.assertDatasetProduces(
119 self.assertDatasetProduces(choose_fastest, expected_output=list(range(100)))
134 self.assertDatasetProduces(choose_fastest, expected_output=[0] * 1000)
158 self.assertDatasetProduces(
173 self.assertDatasetProduces(
|
| D | map_parallelization_test.py | 71 self.assertDatasetProduces(
85 self.assertDatasetProduces(
100 self.assertDatasetProduces(
120 self.assertDatasetProduces(dataset, expected_output=[2, 3, 4, 5])
137 self.assertDatasetProduces(dataset, expected_output=[1, 1, 2, 1, 2, 3])
153 self.assertDatasetProduces(dataset, expected_output=[1, 1, 2, 1, 2, 3])
|
| D | autotune_buffer_sizes_test.py | 46 self.assertDatasetProduces(dataset, range(1, 51))
58 self.assertDatasetProduces(
71 self.assertDatasetProduces(dataset, range(1, 51))
98 self.assertDatasetProduces(dataset, [[i] for i in range(5, 55)])
109 self.assertDatasetProduces(dataset, range(1, 51))
|
| D | choose_fastest_dataset_test.py | 38 self.assertDatasetProduces(
47 self.assertDatasetProduces(
59 self.assertDatasetProduces(
96 self.assertDatasetProduces(
|