Ejemplos de _WrappedSession en Python

Ejemplo n.º 1

Archivo: monitored_session_test.py Proyecto: xordlxordl/tensorflow

 def test_run(self):
     with self.test_session() as sess:
         c = tf.constant(0)
         v = tf.identity(c)
         self.assertEqual(42, sess.run(v, feed_dict={c: 42}))
         wrapped_sess = monitored_session._WrappedSession(sess)
         self.assertEqual(51, wrapped_sess.run(v, feed_dict={c: 51}))

Ejemplo n.º 2

Archivo: monitored_session_test.py Proyecto: JamesFysh/tensorflow

 def test_run(self):
   with self.test_session() as sess:
     c = tf.constant(0)
     v = tf.identity(c)
     self.assertEqual(42, sess.run(v, feed_dict={c: 42}))
     wrapped_sess = monitored_session._WrappedSession(sess)
     self.assertEqual(51, wrapped_sess.run(v, feed_dict={c: 51}))

Ejemplo n.º 3

Archivo: monitored_session_test.py Proyecto: xordlxordl/tensorflow

 def test_close_twice(self):
     with self.test_session() as sess:
         wrapped_sess = monitored_session._WrappedSession(sess)
         wrapped_sess.close()
         self.assertTrue(wrapped_sess.should_stop())
         wrapped_sess.close()
         self.assertTrue(wrapped_sess.should_stop())

Ejemplo n.º 4

Archivo: monitored_session_test.py Proyecto: JamesFysh/tensorflow

 def test_close_twice(self):
   with self.test_session() as sess:
     wrapped_sess = monitored_session._WrappedSession(sess)
     wrapped_sess.close()
     self.assertTrue(wrapped_sess.should_stop())
     wrapped_sess.close()
     self.assertTrue(wrapped_sess.should_stop())

Ejemplo n.º 5

Archivo: monitored_session_test.py Proyecto: xordlxordl/tensorflow

 def test_should_stop_delegates_to_wrapped_session(self):
     with self.test_session() as sess:
         wrapped_sess0 = StopAtNSession(sess, 4)
         wrapped_sess1 = monitored_session._WrappedSession(wrapped_sess0)
         self.assertFalse(wrapped_sess1.should_stop())
         self.assertFalse(wrapped_sess1.should_stop())
         self.assertFalse(wrapped_sess1.should_stop())
         self.assertFalse(wrapped_sess1.should_stop())
         self.assertTrue(wrapped_sess1.should_stop())

Ejemplo n.º 6

Archivo: monitored_session_test.py Proyecto: JamesFysh/tensorflow

 def test_should_stop_delegates_to_wrapped_session(self):
   with self.test_session() as sess:
     wrapped_sess0 = StopAtNSession(sess, 4)
     wrapped_sess1 = monitored_session._WrappedSession(wrapped_sess0)
     self.assertFalse(wrapped_sess1.should_stop())
     self.assertFalse(wrapped_sess1.should_stop())
     self.assertFalse(wrapped_sess1.should_stop())
     self.assertFalse(wrapped_sess1.should_stop())
     self.assertTrue(wrapped_sess1.should_stop())

Ejemplo n.º 7

Archivo: monitored_session_test.py Proyecto: xordlxordl/tensorflow

 def test_properties(self):
     with self.test_session() as sess:
         tf.constant(0.0)
         wrapped_sess = monitored_session._WrappedSession(sess)
         self.assertEquals(sess.graph, wrapped_sess.graph)
         self.assertEquals(sess.sess_str, wrapped_sess.sess_str)

Ejemplo n.º 8

Archivo: monitored_session_test.py Proyecto: JamesFysh/tensorflow

 def test_properties(self):
   with self.test_session() as sess:
     tf.constant(0.0)
     wrapped_sess = monitored_session._WrappedSession(sess)
     self.assertEquals(sess.graph, wrapped_sess.graph)
     self.assertEquals(sess.sess_str, wrapped_sess.sess_str)