| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils_test.py | 37 def _create_checkpoints(sess, checkpoint_dir): argument
38 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
56 def _create_partition_checkpoints(sess, checkpoint_dir): argument
57 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
80 checkpoint_dir = self.get_temp_dir() + "/no_checkpoints"
83 checkpoint_utils.load_variable(checkpoint_dir, "var1"), [])
86 checkpoint_dir = self.get_temp_dir()
88 _, _, _, _ = _create_checkpoints(session, checkpoint_dir)
91 checkpoint_utils.load_variable(checkpoint_dir, "var5"), [])
94 checkpoint_dir = self.get_temp_dir()
[all …]
|
| D | session_manager_test.py | 74 checkpoint_dir = os.path.join(self.get_temp_dir(), "prepare_session")
77 gfile.DeleteRecursively(checkpoint_dir)
81 gfile.MakeDirs(checkpoint_dir)
92 checkpoint_dir=checkpoint_dir)
94 checkpoint_filename = os.path.join(checkpoint_dir,
100 os.rename(checkpoint_dir, checkpoint_dir2)
101 gfile.MakeDirs(checkpoint_dir)
115 checkpoint_dir=checkpoint_dir,
119 gfile.DeleteRecursively(checkpoint_dir)
120 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)
123 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
135 checkpoint_dir = os.path.join(self.get_temp_dir(),
139 self._train_model(checkpoint_dir, num_steps=300)
156 checkpoint_path = saver.latest_checkpoint(checkpoint_dir)
175 checkpoint_dir = os.path.join(self.get_temp_dir(), 'eval_ops_and_final_ops')
178 self._train_model(checkpoint_dir, num_steps=1)
[all …]
|
| D | session_manager.py | 158 checkpoint_dir=None, argument
195 if checkpoint_dir and checkpoint_filename_with_path:
200 if not saver or not (checkpoint_dir or checkpoint_filename_with_path):
209 ckpt = checkpoint_management.get_checkpoint_state(checkpoint_dir)
215 ckpt = checkpoint_management.get_checkpoint_state(checkpoint_dir)
228 checkpoint_dir=None, argument
286 checkpoint_dir=checkpoint_dir,
319 checkpoint_dir=None, argument
351 checkpoint_dir=checkpoint_dir,
364 restoring_file = checkpoint_dir or checkpoint_filename_with_path
|
| D | monitored_session.py | 310 checkpoint_dir=None, argument
355 summary_dir = summary_dir or checkpoint_dir
386 checkpoint_dir):
390 checkpoint_dir,
397 os.path.join(checkpoint_dir, tmpdir),
406 checkpoint_dir=checkpoint_dir,
418 checkpoint_dir=None, argument
510 checkpoint_dir=checkpoint_dir,
539 checkpoint_dir=checkpoint_dir,
543 summary_dir = summary_dir or checkpoint_dir
[all …]
|
| D | checkpoint_management.py | 246 def get_checkpoint_state(checkpoint_dir, latest_filename=None): argument
265 coord_checkpoint_filename = _GetCheckpointFilename(checkpoint_dir,
278 + checkpoint_dir)
282 ckpt.model_checkpoint_path = os.path.join(checkpoint_dir,
287 ckpt.all_model_checkpoint_paths[i] = os.path.join(checkpoint_dir, p)
323 def latest_checkpoint(checkpoint_dir, latest_filename=None): argument
336 ckpt = get_checkpoint_state(checkpoint_dir, latest_filename)
|
| D | monitored_session_test.py | 280 is_chief=True, checkpoint_dir=logdir) as session:
286 is_chief=True, checkpoint_dir=logdir) as session:
296 checkpoint_dir=logdir,
303 is_chief=True, checkpoint_dir=logdir) as session:
313 checkpoint_dir=logdir,
322 is_chief=True, checkpoint_dir=logdir) as session:
333 checkpoint_dir=logdir,
351 checkpoint_dir=logdir,
372 checkpoint_dir=logdir,
383 is_chief=True, checkpoint_dir=logdir) as session:
[all …]
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | checkpoint_utils_test.py | 35 def _create_checkpoints(sess, checkpoint_dir): argument
36 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
54 def _create_partition_checkpoints(sess, checkpoint_dir): argument
55 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
78 checkpoint_dir = self.get_temp_dir() + "/no_checkpoints"
81 checkpoint_utils.load_variable(checkpoint_dir, "var1"), [])
84 checkpoint_dir = self.get_temp_dir()
86 _, _, _, _ = _create_checkpoints(session, checkpoint_dir)
89 checkpoint_utils.load_variable(checkpoint_dir, "var5"), [])
92 checkpoint_dir = self.get_temp_dir()
[all …]
|
| D | checkpoint_utils.py | 66 def load_variable(checkpoint_dir, name): argument
79 reader = load_checkpoint(checkpoint_dir)
83 def list_variables(checkpoint_dir): argument
92 reader = load_checkpoint(checkpoint_dir)
152 def init_from_checkpoint(checkpoint_dir, assignment_map): argument
219 filepattern = _get_checkpoint_filename(checkpoint_dir)
220 reader = load_checkpoint(checkpoint_dir)
242 tensor_name_in_ckpt, checkpoint_dir, variable_map
259 var_name, checkpoint_dir, tensor_name_in_ckpt
292 tensor_name_in_ckpt, checkpoint_dir
[all …]
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | evaluation_test.py | 55 checkpoint_dir = os.path.join(self.get_temp_dir(), 'no_checkpoints_found')
58 for _ in evaluation.checkpoints_iterator(checkpoint_dir, timeout=0):
63 checkpoint_dir = os.path.join(self.get_temp_dir(), 'one_checkpoint_found')
64 if not gfile.Exists(checkpoint_dir):
65 gfile.MakeDirs(checkpoint_dir)
72 save_path = os.path.join(checkpoint_dir, 'model.ckpt')
76 for _ in evaluation.checkpoints_iterator(checkpoint_dir, timeout=0):
81 checkpoint_dir = os.path.join(self.get_temp_dir(),
83 if not gfile.Exists(checkpoint_dir):
84 gfile.MakeDirs(checkpoint_dir)
[all …]
|
| D | evaluation.py | 171 def wait_for_new_checkpoint(checkpoint_dir, argument
189 logging.info('Waiting for new checkpoint at %s', checkpoint_dir)
192 checkpoint_path = checkpoint_management.latest_checkpoint(checkpoint_dir)
202 def checkpoints_iterator(checkpoint_dir, argument
247 checkpoint_dir, checkpoint_path, timeout=timeout)
345 def evaluate_repeatedly(checkpoint_dir, argument
435 checkpoint_dir,
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | checkpoint_utils_test.py | 40 def _create_checkpoints(sess, checkpoint_dir): argument
41 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
60 checkpoint_dir = self.get_temp_dir()
62 v1, v2 = _create_checkpoints(session, checkpoint_dir)
63 return checkpoint_dir, v1, v2
75 checkpoint_dir, v1_value, v2_value = self._get_test_object()
83 checkpoint_utils.init_from_checkpoint(checkpoint_dir, {
110 checkpoint_dir, v1_value, _ = self._get_test_object()
118 checkpoint_utils.init_from_checkpoint(checkpoint_dir, {
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | strip_pruning_vars.py | 43 def strip_pruning_vars(checkpoint_dir, output_node_names, output_dir, filename): argument
67 checkpoint_dir, output_node_names)
77 return strip_pruning_vars(FLAGS.checkpoint_dir, FLAGS.output_node_names,
|
| D | strip_pruning_vars_lib.py | 113 def graph_def_from_checkpoint(checkpoint_dir, output_node_names): argument
129 checkpoint_path = saver_lib.latest_checkpoint(checkpoint_dir)
132 .format(checkpoint_dir))
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| D | core_estimator_predictor.py | 73 checkpoint_dir = estimator.model_dir
77 checkpoint_dir=checkpoint_dir))
|
| /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.train.-checkpoint-saver-hook.pbtxt | 8 …argspec: "args=[\'self\', \'checkpoint_dir\', \'save_secs\', \'save_steps\', \'saver\', \'checkpoi…
|
| D | tensorflow.estimator.-checkpoint-saver-hook.pbtxt | 8 …argspec: "args=[\'self\', \'checkpoint_dir\', \'save_secs\', \'save_steps\', \'saver\', \'checkpoi…
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint.py | 43 checkpoint_dir, argument
73 self._checkpoint_dir = checkpoint_dir
74 self._save_path = os.path.join(checkpoint_dir, checkpoint_basename)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | evaluation.py | 221 checkpoint_dir, argument
297 checkpoint_dir,
|
| /external/tensorflow/tensorflow/examples/tf2_showcase/ |
| D | mnist.py | 209 checkpoint_dir = os.path.join(flags_obj.model_dir, 'checkpoints')
210 checkpoint_prefix = os.path.join(checkpoint_dir, 'ckpt')
215 checkpoint.restore(tf.train.latest_checkpoint(checkpoint_dir))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.train.pbtxt | 77 …argspec: "args=[\'checkpoint_dir\', \'latest_filename\'], varargs=None, keywords=None, defaults=[\…
81 …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/contrib/eager/python/examples/gan/ |
| D | mnist.py | 289 checkpoint_prefix = os.path.join(FLAGS.checkpoint_dir, 'ckpt')
290 latest_cpkt = tf.train.latest_checkpoint(FLAGS.checkpoint_dir)
|