Home
last modified time | relevance | path

Searched refs:get_next (Results 1 – 25 of 148) sorted by relevance

123456

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dsql_dataset_test.py53 get_next = self.getNext(
62 self.evaluate(get_next()))
64 self.evaluate(get_next())
70 get_next = self.getNext(
76 self.assertEqual((b"John", b"Doe", b"n\0nsense"), self.evaluate(get_next()))
78 self.evaluate(get_next()))
80 self.evaluate(get_next())
87 get_next = self.getNext(
92 self.assertEqual((b"John", b"Doe", b"Hi!"), self.evaluate(get_next()))
93 self.assertEqual((b"Jane", b"Moe", b"Hi again!"), self.evaluate(get_next()))
[all …]
Dignore_errors_test.py51 get_next = self.getNext(dataset)
54 self.assertEqual(x, self.evaluate(get_next()))
56 self.evaluate(get_next())
65 get_next = self.getNext(dataset)
67 self.assertEqual(x, self.evaluate(get_next()))
69 self.assertEqual(5., self.evaluate(get_next()))
73 self.evaluate(get_next())
83 get_next = self.getNext(dataset)
86 self.assertEqual(x, self.evaluate(get_next()))
88 self.evaluate(get_next())
[all …]
Dmap_and_batch_test.py76 get_next = self.getNext(dataset)
83 result = self.evaluate(get_next())
89 self.evaluate(get_next())
93 get_next = self.getNext(dataset_fn(8, 14))
98 result = self.evaluate(get_next())
104 result = self.evaluate(get_next())
110 self.evaluate(get_next())
160 get_next = iterator._next_internal # pylint: disable=protected-access
163 get_next = iterator.get_next
167 elements.append(get_next)
[all …]
Dgroup_by_window_test.py68 get_next = self.getNext(bucketed_dataset)
70 which_bucket, bucketed_values = self.evaluate(get_next())
99 get_next = self.getNext(bucketed_dataset)
102 which_bucket_even, bucketed_values_even = self.evaluate(get_next())
103 which_bucket_odd, bucketed_values_odd = self.evaluate(get_next())
165 get_next = self.getNext(bucketed_dataset)
168 which_bucket0, bucketed_values_even0 = self.evaluate(get_next())
169 which_bucket1, bucketed_values_even1 = self.evaluate(get_next())
195 get_next = self.getNext(dataset)
199 result = self.evaluate(get_next())
[all …]
Ddense_to_sparse_batch_test.py40 get_next = self.getNext(dataset)
43 results = self.evaluate(get_next())
55 self.evaluate(get_next())
64 get_next = self.getNext(dataset)
67 results = self.evaluate(get_next())
83 self.evaluate(get_next())
100 get_next = self.getNext(dataset_fn([[1]]))
103 self.evaluate(get_next())
106 get_next = self.getNext(dataset_fn(np.int32(range(13))))
109 self.evaluate(get_next())
Dwrap_unwrap_test.py43 get_next = self.getNext(variant_ds, requires_initialization=True)
45 self.assertEqual(i, self.evaluate(get_next()))
61 get_next = iterator.get_next()
66 self.assertEqual(i, self.evaluate(get_next))
Ddense_to_ragged_batch_test.py121 get_next = self.getNext(dataset)
122 rows = [nest.map_structure(_to_list, self.evaluate(get_next()))
128 get_next = self.getNext(batched_dataset)
134 result = self.evaluate(get_next())
142 self.evaluate(get_next())
163 get_next = self.getNext(dataset)
166 result = self.evaluate(get_next())
Drejection_resample_test.py49 get_next = self.getNext(
59 returned.append(self.evaluate(get_next()))
92 get_next = self.getNext(dataset)
97 returned.append(self.evaluate(get_next()))
123 get_next = self.getNext(dataset)
128 returned.append(self.evaluate(get_next()))
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py82 get_next = iterator.get_next()
85 return init_op, get_next, save_op, restore_op
92 init_op, get_next, save_op, _ = _build_graph(start, stop)
97 self.assertEqual(i, sess.run(get_next))
101 init_op, get_next, _, restore_op = _build_graph(start, stop)
106 self.assertEqual(i, sess.run(get_next))
108 sess.run(get_next)
112 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
117 self.assertEqual(i, sess.run(get_next))
122 self.assertEqual(i, sess.run(get_next))
[all …]
Dfrom_generator_test.py143 get_next = self.getNext(dataset)
146 val0, val1 = self.evaluate(get_next())
150 self.evaluate(get_next())
195 get_next = self.getNext(dataset)
199 self.assertAllEqual(elem, self.evaluate(get_next()))
201 self.evaluate(get_next())
213 get_next = self.getNext(dataset)
216 next_val = self.evaluate(get_next())
220 self.evaluate(get_next())
263 get_next = self.getNext(dataset)
[all …]
Dshuffle_test.py65 get_next = self.getNext(dataset_fn())
68 unshuffled_elements.append(self.evaluate(get_next()))
70 self.evaluate(get_next())
74 get_next = self.getNext(dataset_fn(buffer_size=100, seed=37))
77 shuffled_elements.append(self.evaluate(get_next()))
79 self.evaluate(get_next())
81 self.evaluate(get_next())
85 get_next = self.getNext(dataset_fn(buffer_size=100, seed=37))
88 reshuffled_elements_same_seed.append(self.evaluate(get_next()))
90 self.evaluate(get_next())
[all …]
Dzip_test.py48 get_next = self.getNext(_dataset_factory(components))
50 results = self.evaluate(get_next())
54 self.evaluate(get_next())
56 self.evaluate(get_next())
61 get_next = self.getNext(_dataset_factory(components))
63 results = self.evaluate(get_next())
67 self.evaluate(get_next())
69 self.evaluate(get_next())
90 get_next = self.getNext(dataset)
92 result1, (result2, result3) = self.evaluate(get_next())
[all …]
Dmap_test.py186 get_next = self.getNext(
190 result = self.evaluate(get_next())
194 self.evaluate(get_next())
203 get_next = self.getNext(
210 results.append(sess.run(get_next()))
263 get_next = self.getNext(
268 result = self.evaluate(get_next())
272 self.evaluate(get_next())
291 get_next = self.getNext(
300 results.append(sess.run(get_next()))
[all …]
Diterator_test.py64 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)
147 return iterator.get_next()
[all …]
Diterator_cluster_test.py61 get_next_op = remote_it.get_next()
78 return remote_iterator.get_next()
94 elem = sess.run(iterator_3.get_next())
142 get_next = iterator.get_next()
147 self.assertAllEqual([0, 0, -1, 1, 2], sess.run(get_next))
150 self.assertAllEqual([2, 0], sess.run(get_next))
152 sess.run(get_next)
175 get_next = iterator.get_next()
180 sess.run(get_next)
Dmulti_device_iterator_test.py76 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
80 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
95 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
99 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
114 elements = multi_device_iterator.get_next()
121 elements = multi_device_iterator.get_next()
138 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
141 elem_on_1 = multi_device_iterator.get_next("/cpu:1")
144 elem_on_1, elem_on_2 = multi_device_iterator.get_next()
195 elem_on_1 = multi_device_iterator.get_next("/cpu:1")
[all …]
Dtest_base.py132 get_next = iterator.get_next()
133 return ta_wrapper(lambda: get_next)
149 get_next = self.getNext(
154 results.append(self.evaluate(get_next()))
202 get_next = self.getNext(
205 self.evaluate(get_next())
212 get_next = self.getNext(
216 result.append(self.evaluate(get_next()))
219 self.evaluate(get_next())
221 self.evaluate(get_next())
Dcache_test.py72 get_next = self.getNext(dataset_fn())
77 elements.append(self.evaluate(get_next()))
79 self.evaluate(get_next())
83 get_next = self.getNext(dataset_fn(filename=self.cache_prefix))
86 cached_elements.append(self.evaluate(get_next()))
88 self.evaluate(get_next())
93 get_next = self.getNext(dataset_fn(count=0, filename=self.cache_prefix))
96 replayed_elements.append(self.evaluate(get_next()))
98 self.evaluate(get_next())
103 get_next = self.getNext(
[all …]
Dinterleave_test.py220 get_next = self.getNext(dataset)
225 self.evaluate(get_next())
227 self.assertEqual(value, self.evaluate(get_next()))
229 self.evaluate(get_next())
244 get_next = self.getNext(dataset)
248 self.assertAllEqual(expected, self.evaluate(get_next()))
250 self.evaluate(get_next())
252 self.evaluate(get_next())
293 get_next = self.getNext(dataset)
296 actual_output.append(self.evaluate(get_next()))
Dconcatenate_test.py54 get_next = self.getNext(concatenated)
57 result = self.evaluate(get_next())
66 self.evaluate(get_next())
86 get_next = self.getNext(concatenated)
88 result = self.evaluate(get_next())
97 self.evaluate(get_next())
Dfrom_tensor_slices_test.py48 get_next = self.getNext(dataset)
55 results = self.evaluate(get_next())
59 results = self.evaluate(get_next())
149 get_next = self.getNext(dataset)
182 results = self.evaluate(get_next())
187 self.evaluate(get_next())
193 get_next = self.getNext(dataset)
203 results = self.evaluate(get_next())
207 self.evaluate(get_next())
240 get_next = self.getNext(dataset)
[all …]
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py52 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)
153 get_next = iterator.get_next()
155 train_op.append(get_next)
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Drange_dataset_serialization_test.py69 get_next = iterator.get_next()
72 return init_op, get_next, save_op, restore_op
79 init_op, get_next, save_op, _ = _build_graph(start, stop)
84 self.assertEqual(i, self.evaluate(get_next))
88 init_op, get_next, _, restore_op = _build_graph(start, stop)
93 self.assertEqual(i, self.evaluate(get_next))
95 self.evaluate(get_next)
99 init_op, get_next, save_op, restore_op = _build_graph(start, stop)
104 self.assertEqual(i, self.evaluate(get_next))
108 self.assertEqual(i, self.evaluate(get_next))
[all …]
/external/tensorflow/tensorflow/python/tpu/
Ddatasets_test.py78 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/python/data/experimental/kernel_tests/optimization/
Dshuffle_and_repeat_fusion_test.py42 get_next = self.getNext(dataset)
47 results.append(self.evaluate(get_next()))
50 self.evaluate(get_next())
52 self.evaluate(get_next())

123456