| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| sql_dataset_op_test.py | 39 get_next = iterator.get_next()
40 return init_op, get_next
97 init_op, get_next = self._createSqlDataset((dtypes.string, dtypes.string,
108 self.assertEqual((b"John", b"Doe", b"Hi!"), sess.run(get_next))
109 self.assertEqual((b"Jane", b"Moe", b"Hi again!"), sess.run(get_next))
111 sess.run(get_next)
115 init_op, get_next = self._createSqlDataset((dtypes.string, dtypes.string,
127 self.assertEqual((b"John", b"California", b"Hi!"), sess.run(get_next))
129 sess.run(get_next)
[all...] |
| range_dataset_op_test.py | 47 get_next = iterator.get_next()
49 self.assertEqual(dtypes.int64, get_next[0].dtype)
50 self.assertEqual((), get_next[0].shape)
52 [t.shape for t in get_next[1]])
56 self.assertEqual((20, (b"a", 1, 37.0)), sess.run(get_next))
57 self.assertEqual((21, (b"b", 2, 38.0)), sess.run(get_next))
60 sess.run(get_next)
66 get_next = iterator.get_next()
[all...] |
| bucketing_test.py | 46 get_next = iterator.get_next()
53 result = sess.run(get_next)
73 get_next = iterator.get_next()
82 self.assertAllEqual([0, 0, 0, 0], sess.run(get_next))
83 self.assertAllEqual([1, 1, 1, 1], sess.run(get_next))
84 self.assertAllEqual([2, 2, 2, 2], sess.run(get_next))
85 self.assertAllEqual([0, 0, 0, 0], sess.run(get_next))
94 get_next = iterator.get_next(
[all...] |
| batch_dataset_op_test.py | 52 get_next = iterator.get_next()
58 results = sess.run(get_next)
70 sess.run(get_next)
80 get_next = iterator.get_next()
86 results = sess.run(get_next)
103 sess.run(get_next)
118 get_next = iterator.get_next()
[all...] |
| map_dataset_op_test.py | 53 get_next = iterator.get_next()
58 self.assertEqual(x, sess.run(get_next))
60 sess.run(get_next)
71 get_next = iterator.get_next()
76 self.assertEqual(x, sess.run(get_next))
78 sess.run(get_next)
95 get_next = iterator.get_next()
[all...] |
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| sequence_dataset_op_test.py | 42 get_next = iterator.get_next()
45 [t.shape for t in get_next])
51 results = sess.run(get_next)
56 sess.run(get_next)
61 results = sess.run(get_next)
65 sess.run(get_next)
70 sess.run(get_next)
77 results = sess.run(get_next)
88 get_next = iterator.get_next(
[all...] |
| dataset_from_generator_op_test.py | 39 get_next = iterator.get_next()
46 self.assertAllEqual(elem, sess.run(get_next))
48 sess.run(get_next)
56 get_next = iterator.get_next()
61 self.assertAllEqual(elem, sess.run(get_next))
63 sess.run(get_next)
126 get_next = iterator.get_next()
[all...] |
| range_dataset_op_test.py | 50 get_next = iterator.get_next()
55 self.assertEqual(i, sess.run(get_next))
57 sess.run(get_next)
65 get_next = iterator.get_next()
70 self.assertEqual(i, sess.run(get_next))
72 sess.run(get_next)
81 get_next = iterator.get_next()
[all...] |
| map_dataset_op_test.py | 66 get_next = iterator.get_next()
69 [t.shape for t in get_next])
76 result = sess.run(get_next)
80 sess.run(get_next)
88 results.append(sess.run(get_next))
132 get_next = iterator.get_next()
135 [t.shape for t in get_next])
146 result = sess.run(get_next)
[all...] |
| filter_dataset_op_test.py | 53 get_next = iterator.get_next()
56 [t.shape for t in get_next])
65 result = sess.run(get_next)
69 sess.run(get_next)
81 get_next = iterator.get_next()
84 self.assertEqual(0, sess.run(get_next))
85 self.assertEqual(1, sess.run(get_next))
86 self.assertEqual(3, sess.run(get_next))
[all...] |
| zip_dataset_op_test.py | 46 get_next = iterator.get_next()
57 results = sess.run(get_next)
62 sess.run(get_next)
68 results = sess.run(get_next)
73 sess.run(get_next)
90 get_next = iterator.get_next()
92 self.assertEqual([20], get_next[0].shape)
93 self.assertEqual([22], get_next[1][0].shape
[all...] |
| flat_map_dataset_op_test.py | 44 get_next = iterator.get_next()
50 self.assertEqual(i, sess.run(get_next))
52 sess.run(get_next)
63 get_next = iterator.get_next()
70 self.assertEqual(i, sess.run(get_next))
73 sess.run(get_next)
85 get_next = iterator.get_next()
[all...] |
| shard_dataset_op_test.py | 32 self.assertEqual(2, sess.run(iterator.get_next()))
33 self.assertEqual(7, sess.run(iterator.get_next()))
35 sess.run(iterator.get_next())
44 self.assertEqual((2, 8), sess.run(iterator.get_next()))
45 self.assertEqual((7, 3), sess.run(iterator.get_next()))
47 sess.run(iterator.get_next())
54 self.assertEqual(0, sess.run(iterator.get_next()))
55 self.assertEqual(5, sess.run(iterator.get_next()))
57 sess.run(iterator.get_next())
81 sess.run(iterator.get_next())
[all...] |
| shuffle_dataset_op_test.py | 61 get_next = iterator.get_next()
68 unshuffled_elements.append(sess.run(get_next))
70 sess.run(get_next)
80 shuffled_elements.append(sess.run(get_next))
82 sess.run(get_next)
93 reshuffled_elements_same_seed.append(sess.run(get_next))
95 sess.run(get_next)
106 reshuffled_elements_different_seed.append(sess.run(get_next))
108 sess.run(get_next)
[all...] |
| cache_dataset_op_test.py | 69 get_next = iterator.get_next()
76 elements.append(sess.run(get_next))
78 sess.run(get_next)
86 cached_elements.append(sess.run(get_next))
88 sess.run(get_next)
101 replayed_elements.append(sess.run(get_next))
103 sess.run(get_next)
115 sess.run(get_next)
132 get_next1 = iterator1.get_next()
[all...] |
| batch_dataset_op_test.py | 58 get_next = iterator.get_next()
61 [t.shape.as_list() for t in get_next])
69 result = sess.run(get_next)
75 sess.run(get_next)
84 result = sess.run(get_next)
89 result = sess.run(get_next)
95 sess.run(get_next)
100 sess.run(get_next)
120 get_next = iterator.get_next(
[all...] |
| iterator_ops_test.py | 56 value = dataset.make_one_shot_iterator().get_next()
83 get_next = iterator.get_next()
86 [t.shape for t in get_next])
91 result = sess.run(get_next)
95 sess.run(get_next)
108 get_next = iterator.get_next()
111 [t.shape for t in get_next])
116 result = sess.run(get_next)
[all...] |
| prefetch_dataset_op_test.py | 34 get_next = iterator.get_next()
39 self.assertEqual(m, sess.run(get_next))
41 sess.run(get_next)
|
| dataset_constructor_op_test.py | 48 get_next = iterator.get_next()
51 [t.shape for t in get_next])
55 results = sess.run(get_next)
59 sess.run(get_next)
81 get_next = iterator.get_next()
89 results = sess.run(get_next)
93 sess.run(get_next)
111 get_next = iterator.get_next(
[all...] |
| iterator_ops_cluster_test.py | 56 get_next_op = remote_it.get_next()
72 return remote_iterator.get_next()
88 elem = sess.run(iterator_3.get_next())
134 get_next = iterator.get_next()
139 self.assertAllEqual([0, 0, -1, 1, 2], sess.run(get_next))
142 self.assertAllEqual([2, 0], sess.run(get_next))
144 sess.run(get_next)
166 get_next = iterator.get_next()
[all...] |
| list_files_dataset_op_test.py | 50 sess.run(itr.get_next())
65 produced_filenames.append(compat.as_bytes(sess.run(itr.get_next())))
68 sess.run(itr.get_next())
81 sess.run(itr.get_next())
101 produced_filenames.append(compat.as_bytes(sess.run(itr.get_next())))
106 sess.run(itr.get_next())
126 produced_filenames.append(compat.as_bytes(sess.run(itr.get_next())))
130 sess.run(itr.get_next())
150 produced_filenames.append(compat.as_bytes(sess.run(itr.get_next())))
155 sess.run(itr.get_next())
[all...] |
| reader_dataset_ops_test.py | 92 get_next = iterator.get_next()
100 self.assertEqual(self._lineText(0, i), sess.run(get_next))
102 sess.run(get_next)
109 self.assertEqual(self._lineText(1, i), sess.run(get_next))
111 sess.run(get_next)
117 self.assertEqual(self._lineText(j, i), sess.run(get_next))
119 sess.run(get_next)
126 self.assertEqual(self._lineText(j, i), sess.run(get_next))
128 sess.run(get_next)
[all...] |
| /external/tensorflow/tensorflow/contrib/kafka/python/kernel_tests/ |
| kafka_test.py | 55 get_next = iterator.get_next()
61 self.assertEqual("D" + str(i), sess.run(get_next))
63 sess.run(get_next)
68 self.assertEqual("D" + str(i + 5), sess.run(get_next))
70 sess.run(get_next)
81 self.assertEqual("D" + str(i + j * 5), sess.run(get_next))
83 sess.run(get_next)
95 self.assertEqual("D" + str(i + j * 5), sess.run(get_next))
97 sess.run(get_next)
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| 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(
[all...] |
| /external/tensorflow/tensorflow/examples/get_started/regression/ |
| dnn_regression.py | 48 .repeat().make_one_shot_iterator().get_next())
53 .make_one_shot_iterator().get_next())
|
