| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| monitored_session.py | 29 Scaffold = monitored_session.Scaffold
|
| monitors_test.py | 622 self.scaffold = monitored_session.Scaffold()
646 self.model_dir, save_secs=2, scaffold=self.scaffold)
648 self.scaffold.finalize()
650 sess.run(self.scaffold.init_op)
660 self.model_dir, save_secs=2, scaffold=self.scaffold)
662 self.scaffold.finalize()
664 sess.run(self.scaffold.init_op
[all...] |
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| evaluation.py | 210 scaffold=monitored_session.Scaffold(
299 scaffold=monitored_session.Scaffold(
|
| /external/tensorflow/tensorflow/python/training/ |
| monitored_session.py | 58 @tf_export(v1=['train.Scaffold'])
59 class Scaffold(object):
68 collections in the graph. The `Scaffold` class helps pick these pieces from
71 If you call the scaffold constructor without any arguments, it will pick
73 `scaffold.finalize()` is called. You can pass arguments to the constructor to
77 The following pieces are directly accessible as attributes of the `Scaffold`
102 `init_fn(scaffold, session)`.
116 """Create a scaffold.
123 the init_op. Will be called as `init_fn(scaffold, session)`.
146 copy_from_scaffold: Optional scaffold object to copy fields from. It
[all...] |
| monitored_session_test.py | 58 """Scaffold tests."""
62 scaffold = monitored_session.Scaffold()
63 self.assertEqual(None, scaffold.init_op)
64 self.assertEqual(None, scaffold.init_feed_dict)
65 self.assertEqual(None, scaffold.init_fn)
66 self.assertEqual(None, scaffold.ready_op)
67 self.assertEqual(None, scaffold.ready_for_local_init_op)
68 self.assertEqual(None, scaffold.local_init_op)
69 self.assertEqual(None, scaffold.saver
[all...] |
| training.py | 79 from tensorflow.python.training.monitored_session import Scaffold
|
| basic_session_run_hooks_test.py | 393 self.scaffold = monitored_session.Scaffold()
405 self.scaffold.finalize()
407 sess.run(self.scaffold.init_op)
417 self.model_dir, saver=self.scaffold.saver, scaffold=self.scaffold)
433 self.model_dir, save_secs=2, scaffold=self.scaffold)
435 self.scaffold.finalize(
[all...] |
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| training.py | 129 One can use a tf.Scaffold and an initializing function to do so.
143 scaffold = tf.Scaffold(init_fn=init_fn)
144 tf.contrib.training.train(train_op, my_log_dir, scaffold=scaffold)
174 scaffold = tf.Scaffold(init_fn=init_fn)
177 tf.contrib.training.train(train_op, my_log_dir, scaffold=scaffold)
205 scaffold = tf.Scaffold(init_fn=init_fn
[all...] |
| evaluation.py | 317 def _scaffold_with_init(scaffold, saver, checkpoint_path):
318 """Creates a scaffold that loads the given checkpoint using an init_fn.
321 scaffold: The scaffold to copy.
326 A scaffold with an init_fn that loads the given checkpoint. If the scaffold
327 provided already has an init_fn, the scaffold is returned unchanged.
333 if not scaffold.init_fn:
334 scaffold = monitored_session.Scaffold(
[all...] |
| training_test.py | 436 scaffold=monitored_session.Scaffold(init_fn=init_fn),
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| model_fn_test.py | 62 scaffold=monitored_session.Scaffold())
79 self.assertEqual(model_fn_ops.scaffold, estimator_spec.scaffold)
|
| estimator_test.py | 472 def _init_fn(scaffold, session):
473 _, _ = scaffold, session
483 scaffold=monitored_session.Scaffold(init_fn=_init_fn))
502 scaffold=monitored_session.Scaffold(saver=self.mock_saver))
[all...] |
| estimator.py | [all...] |
| /external/tensorflow/tensorflow/contrib/compiler/ |
| xla_test.py | 548 scaffold = training.Scaffold()
551 return scaffold
|
| /external/tensorflow/tensorflow/python/tpu/ |
| tpu_estimator.py | 289 `scaffold_fn` is a function running on CPU to generate the `Scaffold`. This
366 scaffold = self.scaffold_fn() if self.scaffold_fn else None
374 scaffold=scaffold,
[all...] |
