Home
last modified time | relevance | path

Searched refs:mon_sess (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks_test.py158 mon_sess = monitored_session._HookedSession(sess, [h])
161 mon_sess.run(no_op)
162 self.assertFalse(mon_sess.should_stop())
164 mon_sess.run(no_op)
165 self.assertFalse(mon_sess.should_stop())
167 mon_sess.run(no_op)
168 self.assertTrue(mon_sess.should_stop())
170 mon_sess._should_stop = False
171 mon_sess.run(no_op)
172 self.assertTrue(mon_sess.should_stop())
[all …]
Dmonitored_session_test.py1224 mon_sess = monitored_session._HookedSession(sess=mock_run, hooks=[])
1227 output = mon_sess.run(fetches=a_tensor,
1242 mon_sess = monitored_session._HookedSession(
1246 mon_sess.run(a_tensor)
1267 mon_sess = monitored_session._HookedSession(
1272 mon_sess.run(fetches='a_tensor')
1273 self.assertFalse(mon_sess.should_stop())
1276 mon_sess.run(fetches='a_tensor')
1277 self.assertTrue(mon_sess.should_stop())
1283 mon_sess = monitored_session._HookedSession(
[all …]
/external/tensorflow/tensorflow/python/debug/wrappers/
Ddisk_usage_test.py90 mon_sess = monitored_session._HookedSession(self.sess, [dumping_hook])
91 mon_sess.run(self.inc_v)
99 mon_sess = monitored_session._HookedSession(self.sess, [dumping_hook])
103 mon_sess.run(self.inc_v)
104 mon_sess.run(self.inc_v)
106 mon_sess.run(self.inc_v)
Ddumping_wrapper_test.py270 mon_sess = monitored_session._HookedSession(self.sess, [dumping_hook])
271 mon_sess.run(self.inc_v)
303 mon_sess = monitored_session._HookedSession(self.sess, [dumping_hook])
305 mon_sess.run(self.inc_v)
341 mon_sess = monitored_session._HookedSession(self.sess, [dumping_hook])
343 mon_sess.run(self.inc_v)
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals_test.py453 mon_sess = monitored_session._HookedSession(sess, [sweep_hook])
457 mon_sess.run(train_op)
468 mon_sess.run(mark_sweep_done)
469 mon_sess.run(train_op)
478 mon_sess.run(mark_sweep_done)
479 mon_sess.run(train_op)
496 mon_sess = monitored_session._HookedSession(sess, [hook])
497 mon_sess.run(train_op)
499 self.assertFalse(mon_sess.should_stop())
500 mon_sess.run(train_op)
[all …]
/external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/
Dcifar10_train.py120 log_device_placement=FLAGS.log_device_placement)) as mon_sess:
121 while not mon_sess.should_stop():
122 mon_sess.run(train_op)
124 mon_sess.run(mask_update_op)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimator.py970 mon_sess = monitored_session.MonitoredSession(
976 with mon_sess:
977 if not mon_sess.should_stop():
978 return mon_sess.run(predictions, feed_fn() if feed_fn else None)
980 return self._predict_generator(mon_sess, predictions, feed_fn,
983 def _predict_generator(self, mon_sess, predictions, feed_fn, iterate_batches): argument
984 with mon_sess:
985 while not mon_sess.should_stop():
986 preds = mon_sess.run(predictions, feed_fn() if feed_fn else None)
1000 if self._is_input_constant(feed_fn, mon_sess.graph):
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dtrainer_hooks_test.py67 mon_sess = monitored_session._HookedSession(sess, [hook])
68 mon_sess.run(train_op)
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmonitors_test.py800 mon_sess = monitored_session._HookedSession(sess=sess, hooks=[hook])
802 mon_sess.run(inc_5)
812 mon_sess.run(inc_5)
835 mon_sess = monitored_session._HookedSession(sess=sess, hooks=[hook])
844 output = mon_sess.run(a_tensor)
/external/tensorflow/tensorflow/contrib/model_pruning/
DREADME.md112 with tf.train.MonitoredTrainingSession(...) as mon_sess:
114 mon_sess.run(train_op)
117 mon_sess.run(mask_update_op)