| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils_test.py | 44 def _create_checkpoints(sess, checkpoint_dir): argument
45 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
63 def _create_partition_checkpoints(sess, checkpoint_dir): argument
64 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
87 checkpoint_dir = self.get_temp_dir() + "/no_checkpoints"
90 checkpoint_utils.load_variable(checkpoint_dir, "var1"), [])
93 checkpoint_dir = self.get_temp_dir()
95 _, _, _, _ = _create_checkpoints(session, checkpoint_dir)
98 checkpoint_utils.load_variable(checkpoint_dir, "var5"), [])
101 checkpoint_dir = self.get_temp_dir()
[all …]
|
| D | session_manager_test.py | 91 checkpoint_dir = os.path.join(self.get_temp_dir(), "prepare_session")
94 gfile.DeleteRecursively(checkpoint_dir)
98 gfile.MakeDirs(checkpoint_dir)
109 checkpoint_dir=checkpoint_dir)
111 checkpoint_filename = os.path.join(checkpoint_dir,
117 os.rename(checkpoint_dir, checkpoint_dir2)
118 gfile.MakeDirs(checkpoint_dir)
132 checkpoint_dir=checkpoint_dir,
136 gfile.DeleteRecursively(checkpoint_dir)
137 os.rename(checkpoint_dir2, checkpoint_dir)
[all …]
|
| D | evaluation_test.py | 75 def _train_model(self, checkpoint_dir, num_steps): argument
98 checkpoint_dir=checkpoint_dir,
108 checkpoint_dir = os.path.join(self.get_temp_dir(),
112 self._train_model(checkpoint_dir, num_steps=300)
122 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
134 checkpoint_dir = os.path.join(self.get_temp_dir(),
138 self._train_model(checkpoint_dir, num_steps=300)
154 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
173 checkpoint_dir = os.path.join(self.get_temp_dir(), 'eval_ops_and_final_ops')
176 self._train_model(checkpoint_dir, num_steps=1)
[all …]
|
| D | session_manager.py | 163 checkpoint_dir=None, argument
200 if checkpoint_dir and checkpoint_filename_with_path:
205 if not saver or not (checkpoint_dir or checkpoint_filename_with_path):
214 ckpt = checkpoint_management.get_checkpoint_state(checkpoint_dir)
220 ckpt = checkpoint_management.get_checkpoint_state(checkpoint_dir)
233 checkpoint_dir=None, argument
291 checkpoint_dir=checkpoint_dir,
324 checkpoint_dir=None, argument
356 checkpoint_dir=checkpoint_dir,
369 restoring_file = checkpoint_dir or checkpoint_filename_with_path
|
| D | monitored_session.py | 324 checkpoint_dir=None, argument
372 summary_dir = summary_dir or checkpoint_dir
403 checkpoint_dir):
407 checkpoint_dir,
415 os.path.join(checkpoint_dir, tmpdir),
425 checkpoint_dir=checkpoint_dir,
437 checkpoint_dir=None, argument
536 checkpoint_dir=checkpoint_dir,
566 checkpoint_dir=checkpoint_dir,
570 summary_dir = summary_dir or checkpoint_dir
[all …]
|
| D | checkpoint_utils.py | 121 def wait_for_new_checkpoint(checkpoint_dir, argument
139 logging.info("Waiting for new checkpoint at %s", checkpoint_dir)
142 checkpoint_path = checkpoint_management.latest_checkpoint(checkpoint_dir)
153 def checkpoints_iterator(checkpoint_dir, argument
198 checkpoint_dir, checkpoint_path, timeout=timeout)
|
| D | checkpoint_management.py | 252 def get_checkpoint_state(checkpoint_dir, latest_filename=None): argument
271 coord_checkpoint_filename = _GetCheckpointFilename(checkpoint_dir,
284 + checkpoint_dir)
288 ckpt.model_checkpoint_path = os.path.join(checkpoint_dir,
292 ckpt.all_model_checkpoint_paths[i] = os.path.join(checkpoint_dir, p)
328 def latest_checkpoint(checkpoint_dir, latest_filename=None): argument
351 ckpt = get_checkpoint_state(checkpoint_dir, latest_filename)
|
| D | monitored_session_test.py | 296 is_chief=True, checkpoint_dir=logdir) as session:
302 is_chief=True, checkpoint_dir=logdir) as session:
312 checkpoint_dir=logdir,
319 is_chief=True, checkpoint_dir=logdir) as session:
329 checkpoint_dir=logdir,
338 is_chief=True, checkpoint_dir=logdir) as session:
349 checkpoint_dir=logdir,
367 checkpoint_dir=logdir,
388 checkpoint_dir=logdir,
399 is_chief=True, checkpoint_dir=logdir) as session:
[all …]
|
| D | basic_session_run_hooks.py | 517 checkpoint_dir, argument
550 self._checkpoint_dir = checkpoint_dir
551 self._save_path = os.path.join(checkpoint_dir, checkpoint_basename)
|
| D | basic_session_run_hooks_test.py | 480 checkpoint_dir=self.model_dir) as sess:
503 checkpoint_dir=self.model_dir) as sess:
523 checkpoint_dir=self.model_dir) as sess:
539 checkpoint_dir=self.model_dir) as sess2:
555 checkpoint_dir=self.model_dir) as sess:
573 checkpoint_dir=self.model_dir) as sess2:
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | sidecar_evaluator_test.py | 84 checkpoint_dir = self.get_temp_dir()
87 checkpoint, checkpoint_dir, max_to_keep=2)
91 model, data=None, checkpoint_dir=checkpoint_dir, log_dir=None)
107 checkpoint_dir = os.path.join(self.get_temp_dir(), 'ckpt')
109 logging.info('checkpoint_dir = %s, log_dir = %s', checkpoint_dir, log_dir)
113 checkpoint, checkpoint_dir, max_to_keep=2)
116 file_io.list_directory_v2(checkpoint_dir),
126 checkpoint_dir=checkpoint_dir,
137 checkpoint_dir = os.path.join(self.get_temp_dir(), 'checkpoints')
146 filepath=os.path.join(checkpoint_dir, 'ckpt-{epoch}'),
[all …]
|
| D | sidecar_evaluator.py | 111 checkpoint_dir, argument
148 self.checkpoint_dir = checkpoint_dir
168 self.checkpoint_dir):
|
| D | worker_training_state.py | 47 def __init__(self, model, checkpoint_dir): argument
78 directory=os.path.join(checkpoint_dir, 'chief'),
81 checkpoint_dir, self._model.distribute_strategy)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | checkpoint_utils_test.py | 42 def _create_checkpoints(sess, checkpoint_dir): argument
43 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
62 checkpoint_dir = self.get_temp_dir()
64 v1, v2 = _create_checkpoints(session, checkpoint_dir)
65 return checkpoint_dir, v1, v2
78 checkpoint_dir, v1_value, v2_value = self._get_test_object()
86 checkpoint_utils.init_from_checkpoint(checkpoint_dir, {
112 checkpoint_dir, v1_value, _ = self._get_test_object()
120 checkpoint_utils.init_from_checkpoint(checkpoint_dir, {
|
| D | distribute_coordinator.py | 221 checkpoint_dir=None, argument
256 checkpoint_dir=checkpoint_dir,
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | multi_worker_tutorial_test.py | 144 def fn(model_path, checkpoint_dir): argument
186 write_checkpoint_dir = write_filepath(checkpoint_dir, task_type, task_id)
197 if not tf.io.gfile.exists(checkpoint_dir):
203 latest_checkpoint = tf.train.latest_checkpoint(checkpoint_dir)
210 checkpoint_dir = os.path.join(self.get_temp_dir(), 'ckpt')
216 args=(model_path, checkpoint_dir),
246 def proc_func(checkpoint_dir): argument
297 write_checkpoint_dir = write_filepath(checkpoint_dir, task_type,
302 latest_checkpoint = tf.train.latest_checkpoint(checkpoint_dir)
335 checkpoint_dir = os.path.join(self.get_temp_dir(), 'ckpt')
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.train.pbtxt | 81 …argspec: "args=[\'checkpoint_dir\', \'min_interval_secs\', \'timeout\', \'timeout_fn\'], varargs=N…
85 …argspec: "args=[\'checkpoint_dir\', \'latest_filename\'], varargs=None, keywords=None, defaults=[\…
89 …argspec: "args=[\'checkpoint_dir\', \'latest_filename\'], varargs=None, keywords=None, defaults=[\…
|
| D | tensorflow.estimator.-checkpoint-saver-hook.pbtxt | 8 …argspec: "args=[\'self\', \'checkpoint_dir\', \'save_secs\', \'save_steps\', \'saver\', \'checkpoi…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-session-manager.pbtxt | 11 …argspec: "args=[\'self\', \'master\', \'init_op\', \'saver\', \'checkpoint_dir\', \'checkpoint_fil…
15 …argspec: "args=[\'self\', \'master\', \'saver\', \'checkpoint_dir\', \'checkpoint_filename_with_pa…
|
| D | tensorflow.train.-chief-session-creator.pbtxt | 8 …argspec: "args=[\'self\', \'scaffold\', \'master\', \'config\', \'checkpoint_dir\', \'checkpoint_f…
|
| D | tensorflow.estimator.-checkpoint-saver-hook.pbtxt | 8 …argspec: "args=[\'self\', \'checkpoint_dir\', \'save_secs\', \'save_steps\', \'saver\', \'checkpoi…
|
| D | tensorflow.train.-checkpoint-saver-hook.pbtxt | 8 …argspec: "args=[\'self\', \'checkpoint_dir\', \'save_secs\', \'save_steps\', \'saver\', \'checkpoi…
|
| D | tensorflow.train.-singular-monitored-session.pbtxt | 16 …argspec: "args=[\'self\', \'hooks\', \'scaffold\', \'master\', \'config\', \'checkpoint_dir\', \'s…
|
| D | tensorflow.train.pbtxt | 257 …argspec: "args=[\'master\', \'is_chief\', \'checkpoint_dir\', \'scaffold\', \'hooks\', \'chief_onl…
289 …argspec: "args=[\'checkpoint_dir\', \'min_interval_secs\', \'timeout\', \'timeout_fn\'], varargs=N…
325 …argspec: "args=[\'checkpoint_dir\', \'latest_filename\'], varargs=None, keywords=None, defaults=[\…
357 …argspec: "args=[\'checkpoint_dir\', \'latest_filename\'], varargs=None, keywords=None, defaults=[\…
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint.py | 47 checkpoint_dir: Text,
72 save_path = os.path.join(checkpoint_dir, checkpoint_basename)
82 self._checkpoint_dir = checkpoint_dir
|