| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | sql_dataset_test.py | 47 get_next = self.getNext(
56 self.evaluate(get_next()))
58 self.evaluate(get_next())
63 get_next = self.getNext(
69 self.assertEqual((b"John", b"Doe", b"n\0nsense"), self.evaluate(get_next()))
71 self.evaluate(get_next()))
73 self.evaluate(get_next())
79 get_next = self.getNext(
84 self.assertEqual((b"John", b"Doe", b"Hi!"), self.evaluate(get_next()))
85 self.assertEqual((b"Jane", b"Moe", b"Hi again!"), self.evaluate(get_next()))
[all …]
|
| D | ignore_errors_test.py | 49 get_next = self.getNext(dataset)
52 self.assertEqual(x, self.evaluate(get_next()))
54 self.evaluate(get_next())
63 get_next = self.getNext(dataset)
66 self.assertEqual(x, self.evaluate(get_next()))
68 self.evaluate(get_next())
86 get_next = self.getNext(dataset)
90 self.assertEqual(compat.as_bytes(filename), self.evaluate(get_next()))
92 self.evaluate(get_next())
99 get_next = self.getNext(dataset)
[all …]
|
| D | group_by_window_test.py | 67 get_next = self.getNext(bucketed_dataset)
69 which_bucket, bucketed_values = self.evaluate(get_next())
97 get_next = self.getNext(bucketed_dataset)
100 which_bucket_even, bucketed_values_even = self.evaluate(get_next())
101 which_bucket_odd, bucketed_values_odd = self.evaluate(get_next())
162 get_next = self.getNext(bucketed_dataset)
165 which_bucket0, bucketed_values_even0 = self.evaluate(get_next())
166 which_bucket1, bucketed_values_even1 = self.evaluate(get_next())
191 get_next = self.getNext(dataset)
195 result = self.evaluate(get_next())
[all …]
|
| D | map_and_batch_test.py | 85 get_next = self.getNext(dataset)
92 result = self.evaluate(get_next())
98 self.evaluate(get_next())
102 get_next = self.getNext(dataset_fn(8, 14))
107 result = self.evaluate(get_next())
113 result = self.evaluate(get_next())
119 self.evaluate(get_next())
191 get_next = iterator._next_internal # pylint: disable=protected-access
194 get_next = iterator.get_next
198 elements.append(get_next)
[all …]
|
| D | dense_to_sparse_batch_test.py | 39 get_next = self.getNext(dataset)
42 results = self.evaluate(get_next())
54 self.evaluate(get_next())
62 get_next = self.getNext(dataset)
65 results = self.evaluate(get_next())
81 self.evaluate(get_next())
96 get_next = self.getNext(dataset_fn([[1]]))
99 self.evaluate(get_next())
102 get_next = self.getNext(dataset_fn(np.int32(range(13))))
105 self.evaluate(get_next())
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | dataset_checkpoint_test.py | 70 get_next = iterator.get_next()
73 return init_op, get_next, save_op, restore_op
80 init_op, get_next, save_op, _ = _build_graph(start, stop)
85 self.assertEqual(i, sess.run(get_next))
89 init_op, get_next, _, restore_op = _build_graph(start, stop)
94 self.assertEqual(i, sess.run(get_next))
96 sess.run(get_next)
100 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
105 self.assertEqual(i, sess.run(get_next))
109 self.assertEqual(i, sess.run(get_next))
[all …]
|
| D | iterator_checkpoint_test.py | 44 get_next = iterator.get_next if context.executing_eagerly(
45 ) else functools.partial(self.evaluate, iterator.get_next())
47 self.assertAllEqual([1, 4], get_next())
49 self.assertAllEqual([9, 16], get_next())
50 self.assertAllEqual([25, 36], get_next())
52 self.assertAllEqual([9, 16], get_next())
53 self.assertAllEqual([25, 36], get_next())
55 get_next()
65 get_next_1 = iterator_1.get_next if context.executing_eagerly(
66 ) else functools.partial(self.evaluate, iterator_1.get_next())
[all …]
|
| D | shuffle_test.py | 60 get_next = self.getNext(dataset_fn())
63 unshuffled_elements.append(self.evaluate(get_next()))
65 self.evaluate(get_next())
69 get_next = self.getNext(dataset_fn(buffer_size=100, seed=37))
72 shuffled_elements.append(self.evaluate(get_next()))
74 self.evaluate(get_next())
76 self.evaluate(get_next())
80 get_next = self.getNext(dataset_fn(buffer_size=100, seed=37))
83 reshuffled_elements_same_seed.append(self.evaluate(get_next()))
85 self.evaluate(get_next())
[all …]
|
| D | from_generator_test.py | 112 get_next = self.getNext(dataset)
115 val0, val1 = self.evaluate(get_next())
119 self.evaluate(get_next())
164 get_next = self.getNext(dataset)
168 self.assertAllEqual(elem, self.evaluate(get_next()))
170 self.evaluate(get_next())
181 get_next = self.getNext(dataset)
184 next_val = self.evaluate(get_next())
188 self.evaluate(get_next())
211 get_next = self.getNext(dataset)
[all …]
|
| D | map_test.py | 112 get_next = self.getNext(self._buildMapDataset(components, 14))
115 result = self.evaluate(get_next())
119 self.evaluate(get_next())
129 get_next = self.getNext(self._buildMapDataset(components, 18))
135 results.append(sess.run(get_next()))
181 get_next = self.getNext(
186 result = self.evaluate(get_next())
190 self.evaluate(get_next())
208 get_next = self.getNext(
217 results.append(sess.run(get_next()))
[all …]
|
| D | zip_test.py | 48 get_next = self.getNext(dataset_fn(equal_length_components))
50 results = self.evaluate(get_next())
54 self.evaluate(get_next())
56 self.evaluate(get_next())
59 get_next = self.getNext(dataset_fn(variable_length_components))
61 results = self.evaluate(get_next())
66 self.evaluate(get_next())
68 self.evaluate(get_next())
88 get_next = self.getNext(dataset)
90 result1, (result2, result3) = self.evaluate(get_next())
[all …]
|
| D | iterator_test.py | 64 value = dataset_ops.make_one_shot_iterator(dataset).get_next()
92 get_next = iterator.get_next()
95 [t.shape for t in get_next])
100 result = sess.run(get_next)
104 sess.run(get_next)
119 get_next = iterator.get_next()
122 [t.shape for t in get_next])
127 result = sess.run(get_next)
131 sess.run(get_next)
146 return iterator.get_next()
[all …]
|
| D | multi_device_iterator_test.py | 63 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
67 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
82 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
86 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
101 elements = multi_device_iterator.get_next()
108 elements = multi_device_iterator.get_next()
125 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
128 elem_on_1 = multi_device_iterator.get_next("/cpu:1")
131 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
182 elem_on_1 = multi_device_iterator.get_next("/cpu:1")
[all …]
|
| D | iterator_cluster_test.py | 58 get_next_op = remote_it.get_next()
75 return remote_iterator.get_next()
91 elem = sess.run(iterator_3.get_next())
140 get_next = iterator.get_next()
145 self.assertAllEqual([0, 0, -1, 1, 2], sess.run(get_next))
148 self.assertAllEqual([2, 0], sess.run(get_next))
150 sess.run(get_next)
173 get_next = iterator.get_next()
178 sess.run(get_next)
|
| D | cache_test.py | 64 get_next = self.getNext(dataset_fn())
69 elements.append(self.evaluate(get_next()))
71 self.evaluate(get_next())
75 get_next = self.getNext(dataset_fn(filename=self.cache_prefix))
78 cached_elements.append(self.evaluate(get_next()))
80 self.evaluate(get_next())
85 get_next = self.getNext(dataset_fn(count=0, filename=self.cache_prefix))
88 replayed_elements.append(self.evaluate(get_next()))
90 self.evaluate(get_next())
95 get_next = self.getNext(
[all …]
|
| D | interleave_test.py | 176 get_next = self.getNext(dataset)
181 self.evaluate(get_next())
183 self.assertEqual(value, self.evaluate(get_next()))
185 self.evaluate(get_next())
199 get_next = self.getNext(dataset)
203 self.assertAllEqual(expected, self.evaluate(get_next()))
205 self.evaluate(get_next())
207 self.evaluate(get_next())
236 get_next = self.getNext(dataset)
239 actual_output.append(self.evaluate(get_next()))
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | datasets_test.py | 52 get_next = iterator.get_next()
54 train_op.append(get_next)
77 get_next = iterator.get_next()
79 train_op.append(get_next)
113 get_next = iterator.get_next()
115 train_op.append(get_next)
126 get_next = iterator.get_next()
128 train_op.append(get_next)
152 get_next = iterator.get_next()
154 train_op.append(get_next)
[all …]
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| D | assert_element_shape_test.py | 55 get_next = iterator.get_next()
59 sess.run(get_next)
61 sess.run(get_next)
99 get_next = iterator.get_next()
103 sess.run(get_next)
105 sess.run(get_next)
128 get_next = iterator.get_next()
132 sess.run(get_next)
154 get_next = iterator.get_next()
158 sess.run(get_next)
[all …]
|
| D | slide_dataset_op_test.py | 76 get_next = iterator.get_next()
79 [t.shape.as_list() for t in get_next])
93 result = sess.run(get_next)
100 sess.run(get_next)
140 get_next = iterator.get_next()
143 [t.shape.as_list() for t in get_next])
157 result = sess.run(get_next)
164 sess.run(get_next)
213 get_next = iterator.get_next()
219 actual = sess.run(get_next)
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | datasets_test.py | 65 self.assertEqual(0, iterator.get_next().numpy())
66 self.assertEqual(1, iterator.get_next().numpy())
67 self.assertEqual(2, iterator.get_next().numpy())
68 self.assertEqual(3, iterator.get_next().numpy())
70 iterator.get_next()
74 self.assertEqual(0, iterator.get_next().numpy())
75 self.assertEqual(1, iterator.get_next().numpy())
76 self.assertEqual(2, iterator.get_next().numpy())
77 self.assertEqual(3, iterator.get_next().numpy())
79 iterator.get_next()
[all …]
|
| /external/tensorflow/tensorflow/contrib/kafka/python/kernel_tests/ |
| D | kafka_test.py | 57 get_next = iterator.get_next()
63 self.assertEqual("D" + str(i), sess.run(get_next))
65 sess.run(get_next)
70 self.assertEqual("D" + str(i + 5), sess.run(get_next))
72 sess.run(get_next)
83 self.assertEqual("D" + str(i + j * 5), sess.run(get_next))
85 sess.run(get_next)
97 self.assertEqual("D" + str(i + j * 5), sess.run(get_next))
99 sess.run(get_next)
111 sess.run(get_next))
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | range_dataset_serialization_test.py | 62 get_next = iterator.get_next()
65 return init_op, get_next, save_op, restore_op
72 init_op, get_next, save_op, _ = _build_graph(start, stop)
77 self.assertEqual(i, self.evaluate(get_next))
81 init_op, get_next, _, restore_op = _build_graph(start, stop)
86 self.assertEqual(i, self.evaluate(get_next))
88 self.evaluate(get_next)
92 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
97 self.assertEqual(i, self.evaluate(get_next))
101 self.assertEqual(i, self.evaluate(get_next))
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/benchmarks/ |
| D | autotune_benchmark.py | 52 get_next = iterator.get_next()
57 sess.run(get_next.op)
60 sess.run(get_next.op)
91 get_next = iterator.get_next()
96 sess.run(get_next.op)
99 sess.run(get_next.op)
129 get_next = iterator.get_next()
134 sess.run(get_next.op)
137 sess.run(get_next.op)
189 get_next = iterator.get_next()
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | datasets_test.py | 78 get_next = iterator.get_next()
82 retrieved_values.append(compat.as_bytes(self._sess.run(get_next)))
103 get_next = iterator.get_next()
107 retrieved_values.append(compat.as_bytes(self._sess.run(get_next)))
131 get_next = iterator.get_next()
135 retrieved_values.append(compat.as_bytes(self._sess.run(get_next)))
165 get_next = iterator.get_next()
169 retrieved_values.append(compat.as_bytes(self._sess.run(get_next)))
189 get_next = iterator.get_next()
191 retrieved_values = self._sess.run(get_next)
|
| /external/tensorflow/tensorflow/contrib/kinesis/python/kernel_tests/ |
| D | kinesis_test.py | 67 get_next = iterator.get_next()
73 self.assertEqual("D" + str(i), sess.run(get_next))
75 sess.run(get_next)
111 get_next = iterator.get_next()
122 data.append(sess.run(get_next))
131 data.append(sess.run(get_next))
|