Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:sess

63     with self.test_session() as sess:
65 sess.run(init_fifo_op)
68 unshuffled_elements.append(sess.run(get_next))
70 sess.run(get_next)
74 sess.run(
80 shuffled_elements.append(sess.run(get_next))
82 sess.run(get_next)
87 sess.run(
93 reshuffled_elements_same_seed.append(sess.run(get_next))
95 sess.run(get_next)
100 sess.run(
106 reshuffled_elements_different_seed.append(sess.run(get_next))
108 sess.run(get_next)
116 sess.run(
122 reshuffled_elements_small_buffer.append(sess.run(get_next))
124 sess.run(get_next)
129 sess.run(init_shuffle_op, feed_dict={buffer_size_placeholder: 2,
133 sess.run(get_next)
142 with self.test_session() as sess:
146 counts[sess.run(get_next)] += 1
159 with self.test_session() as sess:
160 initial_permutation = sess.run(next_element)
161 self.assertAllEqual(initial_permutation, sess.run(next_element))
162 self.assertAllEqual(initial_permutation, sess.run(next_element))
164 sess.run(next_element)
174 with self.test_session() as sess:
175 initial_permutation = list(sess.run(next_element))
177 next_permutation = list(sess.run(next_element))
182 sess.run(next_element)