| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | checkpoint_callback_manager.cc | 49 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | session_manager.py | 182 checkpoint_dir=None, argument
254 checkpoint_dir=None, argument
345 checkpoint_dir=None, argument
|
| H A D | checkpoint_utils.py | 122 def wait_for_new_checkpoint(checkpoint_dir, argument
154 def checkpoints_iterator(checkpoint_dir, argument
|
| H A D | monitored_session.py | 318 checkpoint_dir=None, argument
431 checkpoint_dir=None, argument
630 checkpoint_dir=None, argument
1128 checkpoint_dir=None, argument
|
| H A D | checkpoint_utils_test.py | 41 def _create_checkpoints(sess, checkpoint_dir): argument
60 def _create_partition_checkpoints(sess, checkpoint_dir): argument
|
| H A D | evaluation_test.py | 71 def _train_model(self, checkpoint_dir, num_steps): argument
|
| H A D | session_manager_test.py | 147 checkpoint_dir=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| H A D | failure_handling.py | 55 def _non_chief_checkpoint_dir(checkpoint_dir, task_id): argument
394 checkpoint_dir=None, argument
533 checkpoint_dir, cluster_resolver): argument
|
| H A D | failure_handler_test.py | 63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
117 checkpoint_dir, argument
|
| H A D | gce_failure_handler_test.py | 63 def _make_checkpoint_manager(checkpoint, checkpoint_dir, cluster_resolver): argument
105 checkpoint_dir, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | checkpoint_management.py | 249 def get_checkpoint_state(checkpoint_dir, latest_filename=None): argument
327 def latest_checkpoint(checkpoint_dir, latest_filename=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | checkpoint_utils_test.py | 38 def _create_checkpoints(sess, checkpoint_dir): argument
|
| H A D | distribute_coordinator.py | 217 checkpoint_dir=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/ |
| H A D | worker_training_state.py | 44 def __init__(self, model, checkpoint_dir): argument
|
| H A D | distribute_coordinator_utils.py | 182 checkpoint_dir=None, argument
|
| /aosp_15_r20/external/libopus/dnn/torch/lossgen/ |
| H A D | train_lossgen.py | 44 checkpoint_dir='checkpoint' variable
|
| /aosp_15_r20/external/libopus/dnn/torch/plc/ |
| H A D | train_plc.py | 42 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| /aosp_15_r20/external/libopus/dnn/torch/fargan/ |
| H A D | train_fargan.py | 43 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| H A D | adv_train_fargan.py | 64 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| /aosp_15_r20/external/libopus/dnn/torch/lpcnet/ |
| H A D | train_lpcnet.py | 126 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| /aosp_15_r20/external/libopus/dnn/torch/rdovae/ |
| H A D | train_rdovae.py | 75 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| /aosp_15_r20/external/libopus/dnn/torch/osce/ |
| H A D | train_vocoder.py | 103 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| H A D | train_model.py | 113 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| H A D | adv_train_vocoder.py | 107 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|
| H A D | adv_train_model.py | 107 checkpoint_dir = os.path.join(args.output, 'checkpoints') variable
|