Home
last modified time | relevance | path

Searched refs:checkpoint_prefix (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dcheckpoint_management.py367 def checkpoint_exists_internal(checkpoint_prefix): argument
381 pathname = _prefix_to_checkpoint_path(checkpoint_prefix,
385 elif file_io.get_matching_files(checkpoint_prefix):
395 def checkpoint_exists(checkpoint_prefix): argument
410 return checkpoint_exists_internal(checkpoint_prefix)
447 for checkpoint_prefix in checkpoint_prefixes:
449 pathname = _prefix_to_checkpoint_path(checkpoint_prefix,
454 match_maybe_append(checkpoint_prefix)
463 def remove_checkpoint(checkpoint_prefix, argument
477 meta_graph_filename(checkpoint_prefix, meta_graph_suffix))
[all …]
Dsaver.py208 def _AddShardedSaveOpsForV2(self, checkpoint_prefix, per_device): argument
256 string_ops.regex_full_match(checkpoint_prefix, "^s3://.*"),
260 [checkpoint_prefix, _SHARDED_SUFFIX])
281 sharded_prefixes, checkpoint_prefix, delete_old_dirs=True)
285 return array_ops.identity(checkpoint_prefix)
1293 checkpoint_prefix = compat.as_text(save_path)
1294 if not checkpoint_management.checkpoint_exists_internal(checkpoint_prefix):
1296 checkpoint_prefix)
1298 logging.info("Restoring parameters from %s", checkpoint_prefix)
Dcheckpoint_ops_test.py46 checkpoint_prefix = os.path.join(self.get_temp_dir(), 'model')
56 saver.save(sess, checkpoint_prefix, global_step=5)
57 self.checkpoint_file = '{}-5'.format(checkpoint_prefix)
Dcheckpoint_utils_test.py45 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
57 checkpoint_prefix,
64 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
78 checkpoint_prefix,
Dsaver_test.py3103 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
3111 save_path = a_saver.save(sess=sess, save_path=checkpoint_prefix)
3126 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
3134 save_path = a_saver.save(sess=sess, save_path=checkpoint_prefix)
/external/tensorflow/tensorflow/python/keras/tests/
Dtracking_util_test.py272 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
288 root.save(file_prefix=checkpoint_prefix)
297 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
327 root.save(file_prefix=checkpoint_prefix, session=session)
399 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
429 root.save(file_prefix=checkpoint_prefix)
453 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
457 checkpoint.save(checkpoint_prefix)
539 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
547 obj.save(checkpoint_prefix)
[all …]
Dtracking_util_with_v1_optimizers_test.py261 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
276 root.save(file_prefix=checkpoint_prefix)
283 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
306 root.save(file_prefix=checkpoint_prefix)
314 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
341 root.save(file_prefix=checkpoint_prefix)
350 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
378 root.save(file_prefix=checkpoint_prefix, session=session)
423 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
455 root.save(file_prefix=checkpoint_prefix)
[all …]
Dsaver_test.py106 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
112 save_path = object_saver.save(file_prefix=checkpoint_prefix)
140 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
146 save_path = object_saver.save(file_prefix=checkpoint_prefix)
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_test.py452 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
453 save_path = original.save(checkpoint_prefix)
484 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
485 save_path = dep_after_var.save(checkpoint_prefix)
653 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
659 save_path = first.save(checkpoint_prefix)
677 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
683 saver.save(checkpoint_prefix)
694 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
704 checkpoint.save(checkpoint_prefix)
[all …]
Dutil_with_v1_optimizers_test.py109 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
115 obj.save(checkpoint_prefix)
117 obj.save(checkpoint_prefix)
126 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
132 save_path = obj.save(checkpoint_prefix)
141 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
174 save_path = second_root_trackable.save(checkpoint_prefix)
242 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
243 save_path = save_root.save(checkpoint_prefix)
/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py197 checkpoint_prefix = "input"
202 checkpoint_prefix += suffix
215 checkpoint_basename=checkpoint_prefix + ".ckpt")
225 self._latest_filename = "checkpoint_" + checkpoint_prefix
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py162 checkpoint_prefix = saved_model_utils.get_variables_path(saved_model_path)
163 model.save_weights(checkpoint_prefix, save_format='tf', overwrite=True)
164 return checkpoint_prefix
426 checkpoint_prefix = os.path.join(
430 model.load_weights(checkpoint_prefix)
/external/tensorflow/tensorflow/python/tools/
Dfreeze_graph_test.py55 checkpoint_prefix = os.path.join(self.get_temp_dir(), "saved_checkpoint")
73 checkpoint_prefix,
174 checkpoint_prefix = os.path.join(tmp_dir, "meta_graph_checkpoint")
189 checkpoint_prefix,
280 checkpoint_prefix = os.path.join(self.get_temp_dir(), "saved_checkpoint")
312 checkpoint_prefix,
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py302 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
309 save_path = checkpoint.save(checkpoint_prefix)
321 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
338 save_path = checkpoint.save(checkpoint_prefix)
350 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
357 save_path = checkpoint.save(checkpoint_prefix)
361 save_path = checkpoint.save(checkpoint_prefix)
369 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
380 checkpoint.save(file_prefix=checkpoint_prefix)
539 checkpoint_prefix = os.path.join(checkpoint_directory, "ckpt")
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dcheckpoint_utils_test.py43 checkpoint_prefix = os.path.join(checkpoint_dir, "model")
52 checkpoint_prefix,
Dtpu_strategy_test.py1183 checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt")
1188 checkpoint.save(file_prefix=checkpoint_prefix)
/external/tensorflow/tensorflow/python/tpu/
Dsession_support.py324 def __init__(self, checkpoint_prefix, saver=None, on_shutdown_hooks=None): argument
326 self._checkpoint_prefix = checkpoint_prefix
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.pbtxt285 argspec: "args=[\'checkpoint_prefix\'], varargs=None, keywords=None, defaults=None"
425 …argspec: "args=[\'checkpoint_prefix\', \'checkpoint_format_version\', \'meta_graph_suffix\'], vara…
/external/tensorflow/tensorflow/python/keras/
Dmetrics_test.py141 checkpoint_prefix = os.path.join(checkpoint_directory, 'ckpt')
151 save_path = checkpoint.save(checkpoint_prefix)
297 checkpoint_prefix = os.path.join(checkpoint_directory, 'ckpt')
307 save_path = checkpoint.save(checkpoint_prefix)