Home | History | Annotate | Download | only in kernel_tests

Lines Matching refs:sess

54     with self.test_session() as sess:
55 sess.run(init_op)
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)
74 with self.test_session() as sess:
75 self.assertEqual(3, sess.run(get_next))
76 self.assertEqual(3 + 4, sess.run(get_next))
77 self.assertEqual(3 + 2 * 4, sess.run(get_next))
79 self.assertEqual(0, sess.run(negative_get_next))
80 self.assertEqual(-1, sess.run(negative_get_next))
81 self.assertEqual(-2, sess.run(negative_get_next))
123 with self.test_session(graph=g) as sess:
124 sess.run(variables.global_variables_initializer())
125 sess.run(init_op)
127 self.assertEqual(i, sess.run(get_next))
128 sess.run(save_op)
132 with self.test_session(graph=g) as sess:
133 sess.run(init_op)
134 sess.run(restore_op)
136 self.assertEqual(i, sess.run(get_next))
138 sess.run(get_next)
143 with self.test_session(graph=g) as sess:
144 sess.run(variables.global_variables_initializer())
145 sess.run(init_op)
147 self.assertEqual(i, sess.run(get_next))
148 sess.run(save_op)
149 sess.run(restore_op)
151 self.assertEqual(i, sess.run(get_next))
153 sess.run(get_next)