Home
last modified time | relevance | path

Searched refs:scaffold (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/training/python/training/
Devaluation.py317 def _scaffold_with_init(scaffold, saver, checkpoint_path): argument
333 if not scaffold.init_fn:
334 scaffold = monitored_session.Scaffold(
335 init_op=scaffold.init_op,
336 init_feed_dict=scaffold.init_feed_dict,
338 ready_op=scaffold.ready_op,
339 local_init_op=scaffold.local_init_op,
340 summary_op=scaffold.summary_op,
341 saver=scaffold.saver)
342 return scaffold
[all …]
Dtraining.py480 scaffold=None, argument
536 scaffold=scaffold,
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks_test.py393 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()
437 sess.run(self.scaffold.init_op)
450 scaffold=self.scaffold,
453 self.scaffold.finalize()
455 sess.run(self.scaffold.init_op)
[all …]
Dmonitored_session.py309 scaffold, argument
372 scaffold=scaffold,
379 scaffold=scaffold,
393 scaffold=scaffold))
400 scaffold=scaffold))
404 scaffold,
419 scaffold=None, argument
503 scaffold = scaffold or Scaffold()
509 scaffold,
525 scaffold=scaffold,
[all …]
Dmonitored_session_test.py62 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)
73 scaffold = monitored_session.Scaffold()
77 scaffold.finalize()
[all …]
Devaluation.py176 scaffold=None, argument
261 scaffold=scaffold,
Dbasic_session_run_hooks.py508 scaffold=None, argument
528 if saver is not None and scaffold is not None:
533 self._scaffold = scaffold
769 scaffold=None, argument
791 if ((scaffold is None and summary_op is None) or
792 (scaffold is not None and summary_op is not None)):
798 self._scaffold = scaffold
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmonitors_test.py622 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)
693 self.model_dir, save_steps=2, scaffold=self.scaffold)
695 self.scaffold.finalize()
697 sess.run(self.scaffold.init_op)
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dmodel_fn.py95 scaffold=None): argument
209 scaffold=scaffold,
307 scaffold=self.scaffold)
Destimator.py935 scaffold=model_fn_results.scaffold,
973 scaffold=infer_ops.scaffold,
1054 scaffold = model_fn_ops.scaffold or monitored_session.Scaffold()
1055 if not (scaffold.saver or ops.get_collection(ops.GraphKeys.SAVERS)):
1080 scaffold=scaffold)
1086 scaffold=scaffold,
1421 if (model_fn_ops.scaffold is not None and
1422 model_fn_ops.scaffold.saver is not None):
1423 saver_for_restore = model_fn_ops.scaffold.saver
Dmodel_fn_test.py62 scaffold=monitored_session.Scaffold())
79 self.assertEqual(model_fn_ops.scaffold, estimator_spec.scaffold)
/external/tensorflow/tensorflow/contrib/compiler/
Dxla.py552 scaffold = scaffold_fn()
553 if scaffold is None:
557 return scaffold
588 scaffold=_get_scaffold(captured_scaffold_fn))
607 scaffold=_get_scaffold(captured_scaffold_fn))
697 if estimator_spec.scaffold:
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py366 scaffold = self.scaffold_fn() if self.scaffold_fn else None
374 scaffold=scaffold,
1727 if estimator_spec.scaffold:
2838 compile_op, loss, host_call, scaffold, training_hooks = (
2925 scaffold=scaffold)
2948 scaffold=scaffold)
2951 compile_op, total_loss, host_calls, scaffold, eval_hooks = (
3029 scaffold=scaffold)
3035 scaffold, prediction_hooks) = _predict_on_tpu_system(
3103 scaffold=scaffold)
[all …]
Dasync_checkpoint.py48 scaffold=None, argument
68 if saver is not None and scaffold is not None:
75 self._scaffold = scaffold
/external/tensorflow/tensorflow/contrib/slim/python/slim/
Devaluation.py210 scaffold=monitored_session.Scaffold(
299 scaffold=monitored_session.Scaffold(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-worker-session-creator.pbtxt8 …argspec: "args=[\'self\', \'scaffold\', \'master\', \'config\', \'max_wait_secs\'], varargs=None, …
Dtensorflow.train.-chief-session-creator.pbtxt8 …argspec: "args=[\'self\', \'scaffold\', \'master\', \'config\', \'checkpoint_dir\', \'checkpoint_f…
Dtensorflow.estimator.-estimator-spec.pbtxt35 name: "scaffold"
Dtensorflow.estimator.-summary-saver-hook.pbtxt8 …', \'save_steps\', \'save_secs\', \'output_dir\', \'summary_writer\', \'scaffold\', \'summary_op\'…
Dtensorflow.train.-summary-saver-hook.pbtxt8 …', \'save_steps\', \'save_secs\', \'output_dir\', \'summary_writer\', \'scaffold\', \'summary_op\'…
Dtensorflow.train.-checkpoint-saver-hook.pbtxt8 …', \'save_secs\', \'save_steps\', \'saver\', \'checkpoint_basename\', \'scaffold\', \'listeners\']…
Dtensorflow.estimator.-checkpoint-saver-hook.pbtxt8 …', \'save_secs\', \'save_steps\', \'saver\', \'checkpoint_basename\', \'scaffold\', \'listeners\']…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.-estimator-spec.pbtxt35 name: "scaffold"
Dtensorflow.estimator.-summary-saver-hook.pbtxt8 …', \'save_steps\', \'save_secs\', \'output_dir\', \'summary_writer\', \'scaffold\', \'summary_op\'…
Dtensorflow.estimator.-checkpoint-saver-hook.pbtxt8 …', \'save_secs\', \'save_steps\', \'saver\', \'checkpoint_basename\', \'scaffold\', \'listeners\']…

12