Home
last modified time | relevance | path

Searched defs:checkpoint_key (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Dtrackable_utils.py136 def checkpoint_key(object_path, local_name): function
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dtrackable_object_graph.proto35 string checkpoint_key = 3; field
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
H A Drestore_ops.cc70 const std::string& checkpoint_key, DataType dtype, in SingleRestore()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/
H A Dtf_saved_model_api.cc144 const std::string& checkpoint_key = attribute->checkpoint_key(); in RestoreCheckpoint() local
/aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/
H A Drestore.py562 def _extract_saveable_name(checkpoint_key): argument
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dtrackable_object_graph.pb.h1259 inline const std::string& TrackableObjectGraph_TrackableObject_SerializedTensor::checkpoint_key() c… in checkpoint_key() function
1290 …Graph_TrackableObject_SerializedTensor::set_allocated_checkpoint_key(std::string* checkpoint_key) { in set_allocated_checkpoint_key()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/saving/
H A Dsaveable_object_util.py415 def save_fn(checkpoint_key): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dimport_model.cc3454 const auto& checkpoint_key = serialized_tensor_attr->checkpoint_key(); in CreateSavedModelIR() local