Searched defs:checkpoint_key (Results 1 – 7 of 7) sorted by relevance
136 def checkpoint_key(object_path, local_name): function
35 string checkpoint_key = 3; field
70 const std::string& checkpoint_key, DataType dtype, in SingleRestore()
144 const std::string& checkpoint_key = attribute->checkpoint_key(); in RestoreCheckpoint() local
562 def _extract_saveable_name(checkpoint_key): argument
415 def save_fn(checkpoint_key): argument
3454 const auto& checkpoint_key = serialized_tensor_attr->checkpoint_key(); in CreateSavedModelIR() local