Home
last modified time | relevance | path

Searched defs:checkpoint_id (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcheckpoint_callback_manager_test.cc87 SaveCallback first_callback = [](absl::string_view checkpoint_id) { in TEST_F()
91 SaveCallback second_callback = [](absl::string_view checkpoint_id) { in TEST_F()
108 RestoreCallback second_callback = [](absl::string_view checkpoint_id, in TEST_F()
122 SaveCallback first_callback = [](absl::string_view checkpoint_id) { in TEST_F()
126 SaveCallback second_callback = [](absl::string_view checkpoint_id) { in TEST_F()
147 RestoreCallback second_callback = [](absl::string_view checkpoint_id, in TEST_F()
165 SaveCallback save_callback1 = [](absl::string_view checkpoint_id) { in TEST_F()
168 SaveCallback save_callback2 = [](absl::string_view checkpoint_id) { in TEST_F()
194 SaveCallback save_callback = [](absl::string_view checkpoint_id) { in TEST_F()
226 absl::string_view str) { in TEST_F()
[all …]
H A Dcheckpoint_callback_manager.cc48 void TriggerSaveCallbackIfFileNotExist(absl::string_view checkpoint_id, in TriggerSaveCallbackIfFileNotExist()
83 void TriggerRestoreCallbackIfFileExists(absl::string_view checkpoint_id, in TriggerRestoreCallbackIfFileExists()
164 std::string checkpoint_id; in RegisterSaveCallback() local
198 std::string checkpoint_id; in RegisterRestoreCallback() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dsnapshot_utils.cc91 uint64 checkpoint_id) { in GetCheckpointFileName()
1034 uint64 checkpoint_id, const std::string& compression, in AsyncWriter()
1070 uint64 checkpoint_id, in WriterThread()
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dfilesystem.py605 def checkpoint_id(self) -> Union[str, os.PathLike]: member in _FileSystemWriter
699 def checkpoint_id(self) -> Union[str, os.PathLike]: member in FileSystemReader
H A Dmetadata.py130 checkpoint_id: Union[str, os.PathLike, None] = None variable in StorageMeta
/aosp_15_r20/external/ltp/testcases/kernel/connectors/pec/
H A Devent_generator.c32 static int checkpoint_id = -1; variable
H A Dpec_listener.c59 static int checkpoint_id = -1; variable