Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dcheckpoint_callback_manager.cc49 absl::string_view checkpoint_dir, in TriggerSaveCallbackIfFileNotExist()
84 absl::string_view checkpoint_dir, in TriggerRestoreCallbackIfFileExists()
165 std::string checkpoint_dir; in RegisterSaveCallback() local
199 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
/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
Dgce_failure_handler_test.py63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
105 checkpoint_dir, argument
Dfailure_handler_test.py63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
117 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
/external/libopus/dnn/torch/lossgen/
Dtrain_lossgen.py44 checkpoint_dir='checkpoint' variable
/external/libopus/dnn/torch/plc/
Dtrain_plc.py42 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
/external/libopus/dnn/torch/fargan/
Dtrain_fargan.py43 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
Dadv_train_fargan.py64 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
/external/libopus/dnn/torch/lpcnet/
Dtrain_lpcnet.py126 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
/external/libopus/dnn/torch/rdovae/
Dtrain_rdovae.py75 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
/external/libopus/dnn/torch/osce/
Dtrain_vocoder.py103 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
Dtrain_model.py113 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
Dadv_train_vocoder.py107 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
Dadv_train_model.py107 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable

12