Home
last modified time | relevance | path

Searched defs:checkpoint_dir (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcheckpoint_callback_manager.cc47 absl::string_view checkpoint_dir, in TriggerSaveCallbackIfFileNotExist()
82 absl::string_view checkpoint_dir, in TriggerRestoreCallbackIfFileExists()
149 std::string checkpoint_dir; in RegisterSaveCallback() local
183 std::string checkpoint_dir; in RegisterRestoreCallback() local
/external/tensorflow/tensorflow/python/training/
Dsession_manager.py182 checkpoint_dir=None, argument
254 checkpoint_dir=None, argument
345 checkpoint_dir=None, argument
Dcheckpoint_utils.py122 def wait_for_new_checkpoint(checkpoint_dir, argument
154 def checkpoints_iterator(checkpoint_dir, argument
Dcheckpoint_utils_test.py41 def _create_checkpoints(sess, checkpoint_dir): argument
60 def _create_partition_checkpoints(sess, checkpoint_dir): argument
Devaluation_test.py71 def _train_model(self, checkpoint_dir, num_steps): argument
Dmonitored_session.py318 checkpoint_dir=None, argument
431 checkpoint_dir=None, argument
630 checkpoint_dir=None, argument
1128 checkpoint_dir=None, argument
Dsession_manager_test.py147 checkpoint_dir=None, argument
Dbasic_session_run_hooks.py528 checkpoint_dir, argument
/external/tensorflow/tensorflow/python/distribute/failure_handling/
Dfailure_handling.py55 def _non_chief_checkpoint_dir(checkpoint_dir, task_id): argument
394 checkpoint_dir=None, argument
533 checkpoint_dir, cluster_resolver): argument
Dfailure_handler_test.py63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
117 checkpoint_dir, argument
Dgce_failure_handler_test.py63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
105 checkpoint_dir, argument
/external/tensorflow/tensorflow/python/keras/distribute/
Dworker_training_state.py44 def __init__(self, model, checkpoint_dir): argument
Ddistribute_coordinator_utils.py182 checkpoint_dir=None, argument
/external/tensorflow/tensorflow/python/distribute/
Dcheckpoint_utils_test.py38 def _create_checkpoints(sess, checkpoint_dir): argument
Ddistribute_coordinator.py217 checkpoint_dir=None, argument
/external/tensorflow/tensorflow/python/checkpoint/
Dcheckpoint_management.py249 def get_checkpoint_state(checkpoint_dir, latest_filename=None): argument
327 def latest_checkpoint(checkpoint_dir, latest_filename=None): argument