Home
last modified time | relevance | path

Searched refs:next_element (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcopy_to_device_test.py47 next_element = iterator.get_next()
54 self.assertEqual(dtypes.int64, next_element.dtype)
55 self.assertEqual([], next_element.shape)
60 self.assertEqual(i, self.evaluate(next_element))
62 self.evaluate(next_element)
72 next_element = iterator.get_next()
79 self.assertEqual(dtypes.int32, next_element.dtype)
80 self.assertEqual((4,), next_element.shape)
84 self.assertAllEqual([0, 1, 2, 3], self.evaluate(next_element))
86 self.evaluate(next_element)
[all …]
Dprefetch_to_device_test.py43 next_element = iterator.get_next()
50 self.assertEqual(dtypes.int64, next_element.dtype)
51 self.assertEqual([], next_element.shape)
56 self.assertEqual(i, self.evaluate(next_element))
58 self.evaluate(next_element)
69 next_element = iterator.get_next()
76 self.assertEqual(dtypes.int64, next_element.dtype)
77 self.assertEqual([], next_element.shape)
81 self.assertEqual(i, self.evaluate(next_element))
83 self.evaluate(next_element)
[all …]
Dparallel_interleave_test.py175 next_element = self.getNext(
187 self.evaluate(next_element()))
189 self.evaluate(next_element())
206 next_element = self.getNext(
224 output = self.evaluate(next_element())
227 self.evaluate(next_element())
234 next_element = self.getNext(
248 actual_element = self.evaluate(next_element())
257 self.evaluate(next_element())
276 next_element = self.getNext(
[all …]
Dstats_dataset_ops_test.py68 next_element = self.getNext(dataset, requires_initialization=True)
73 np.array([i] * i, dtype=np.int64), self.evaluate(next_element()))
79 self.evaluate(next_element())
89 next_element = self.getNext(dataset, requires_initialization=True)
92 self.assertEqual(i, self.evaluate(next_element()))
97 self.evaluate(next_element())
106 next_element = self.getNext(dataset, requires_initialization=True)
109 np.array([i] * i, dtype=np.int64), self.evaluate(next_element()))
124 self.evaluate(next_element())
135 next_element = self.getNext(dataset, requires_initialization=True)
[all …]
Dscan_test.py54 next_element = self.getNext(dataset_fn(start_val, step_val, take_val))
57 self.assertEqual(expected, self.evaluate(next_element()))
59 self.evaluate(next_element())
64 next_element = self.getNext(data)
66 self.assertEqual(1, self.evaluate(next_element()))
67 self.assertEqual(1, self.evaluate(next_element()))
68 self.assertEqual(2, self.evaluate(next_element()))
69 self.assertEqual(3, self.evaluate(next_element()))
70 self.assertEqual(5, self.evaluate(next_element()))
71 self.assertEqual(8, self.evaluate(next_element()))
[all …]
Dtake_while_test.py74 next_element = self.getNext(dataset)
75 self.assertEqual(b"this", self.evaluate(next_element()))
76 self.assertEqual(b"is", self.evaluate(next_element()))
77 self.assertEqual(b"the", self.evaluate(next_element()))
80 self.assertEqual(b"test", self.evaluate(next_element()))
93 next_element = self.getNext(dataset)
96 self.assertTrue(self.evaluate(next_element()))
99 self.evaluate(next_element())
Ddirected_interleave_dataset_test.py41 next_element = self.getNext(dataset)
45 self.assertEqual(i, self.evaluate(next_element()))
47 self.evaluate(next_element())
68 next_element = self.getNext(dataset)
71 freqs[self.evaluate(next_element())] += 1
73 self.evaluate(next_element())
102 next_element = self.getNext(dataset)
104 self.assertEqual(words[i], self.evaluate(next_element()))
106 self.evaluate(next_element())
Dsleep_test.py40 next_element = self.getNext(dataset)
43 self.assertEqual(i, self.evaluate(next_element()))
47 self.evaluate(next_element())
Dmake_tf_record_dataset_test.py113 next_element = self.getNext(dataset)
117 first_batches.append(self.evaluate(next_element()))
121 next_element = self.getNext(dataset)
125 second_batches.append(self.evaluate(next_element()))
Dstats_dataset_test_base.py97 next_element = self.getNext(dataset, requires_initialization=True)
100 next_ = self.evaluate(next_element())
114 self.evaluate(next_element())
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dshuffle_test.py160 next_element = self.getNext(dataset)
162 initial_permutation = self.evaluate(next_element())
163 self.assertAllEqual(initial_permutation, self.evaluate(next_element()))
164 self.assertAllEqual(initial_permutation, self.evaluate(next_element()))
166 self.evaluate(next_element())
171 next_element = self.getNext(dataset)
173 initial_permutation = list(self.evaluate(next_element()))
175 next_permutation = list(self.evaluate(next_element()))
179 self.evaluate(next_element())
199 next_element = iterator.get_next()
[all …]
Diterator_test.py173 next_element = iterator.get_next()
180 self.assertAllEqual([1, 4, 9], sess.run(next_element))
182 sess.run(next_element)
190 results.append(sess.run(next_element))
215 next_element = iterator.get_next()
219 sess.run(next_element)
223 sess.run(next_element)
229 sess.run(next_element)
350 next_element = iterator.get_next()
357 self.assertEqual(expected, sess.run(next_element))
[all …]
Dlist_files_test.py93 next_element = self.getNext(dataset, requires_initialization=True)
96 actual_filenames.append(self.evaluate(next_element()))
174 next_element = self.getNext(dataset)
181 actual_filenames.append(compat.as_bytes(self.evaluate(next_element())))
183 self.evaluate(next_element())
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Doptimize_benchmark.py55 next_element = iterator.get_next()
59 sess.run(next_element.op)
64 sess.run(next_element.op)
96 next_element = iterator.get_next()
100 sess.run(next_element.op)
105 sess.run(next_element.op)
137 next_element = iterator.get_next()
141 sess.run(next_element.op)
146 sess.run(next_element.op)
Dchoose_fastest_benchmark.py68 next_element = iterator.get_next()
75 sess.run(next_element.op)
84 next_element = iterator.get_next()
91 sess.run(next_element.op)
96 sess.run(next_element.op)
Dunbatch_benchmark.py49 next_element = iterator.get_next()
59 sess.run(next_element.op)
86 next_element = iterator.get_next()
96 sess.run(next_element.op)
Dparallel_interleave_benchmark.py62 next_element = dataset_ops.make_one_shot_iterator(dataset).get_next()
68 sess.run(next_element.op)
Dmatching_files_benchmark.py66 next_element = dataset_ops.make_one_shot_iterator(dataset).get_next()
73 sess.run(next_element)
/external/tensorflow/tensorflow/python/data/benchmarks/
Dbenchmark_base.py59 next_element = iterator.get_next()
60 next_element = nest.flatten(next_element)[0]
67 sess.run(next_element)
71 sess.run(next_element.op)
Dmeta_benchmark.py114 next_element = iterator.get_next()
120 get_next_element = sess.make_callable(next_element)
125 get_next_element = lambda: sess.run(next_element.op)
139 next_element = iterator.get_next()
146 timeit.timeit(lambda: sess.run(next_element.op), number=1))
Dlist_files_benchmark.py64 next_element = dataset.make_one_shot_iterator().get_next()
70 sess.run(next_element)
/external/tensorflow/tensorflow/python/distribute/
Dinput_ops_test.py94 next_element = iterator.get_next()
98 self.assertAllEqual(record_fn(r, f), self.evaluate(next_element))
100 self.evaluate(next_element)
143 next_element = iterator.get_next()
148 actual.append(self.evaluate(next_element))
151 self.evaluate(next_element)
176 next_element = iterator.get_next()
182 actual.extend(self.evaluate(next_element))
184 self.evaluate(next_element)
214 next_element = iterator.get_next()
[all …]
/external/tensorflow/tensorflow/contrib/distribute/python/
Dinput_lib_test.py75 next_element = iterator.get_next()
77 [values.select_replica(r, next_element) for r in range(len(devices))])
83 next_element = iterator.get_next()
85 [values.select_replica(r, next_element) for r in range(len(devices))])
91 next_element = iterator.get_next()
93 [values.select_replica(r, next_element) for r in range(len(devices))])
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dscan_dataset_op.cc170 std::vector<Tensor> next_element; in GetNextInternal() local
172 input_impl_->GetNext(ctx, &next_element, end_of_sequence)); in GetNextInternal()
178 args.reserve(state_.size() + next_element.size()); in GetNextInternal()
180 std::copy(next_element.begin(), next_element.end(), in GetNextInternal()
/external/tensorflow/tensorflow/python/keras/engine/
Ddistributed_training_utils.py534 next_element = iterator.get_next()
536 if len(nest.flatten(next_element)) == len(model.inputs):
537 x = next_element
540 elif len(nest.flatten(next_element)) == (len(model.inputs) +
542 x, y = next_element
545 x, y, sample_weights = next_element

12