Home
last modified time | relevance | path

Searched refs:checkpoint_key (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dtf_saved_model_api.cc143 const std::string& checkpoint_key = attribute->checkpoint_key(); in RestoreCheckpoint() local
144 if (!bundle->variable_reader()->Contains(checkpoint_key)) { in RestoreCheckpoint()
145 LOG(WARNING) << "No checkpoint entry found for " << checkpoint_key in RestoreCheckpoint()
155 context, variables_path_prefix, checkpoint_key, variable->dtype(), in RestoreCheckpoint()
/external/tensorflow/tensorflow/python/training/tracking/
Dgraph_view.py262 attribute.checkpoint_key = "%s/%s/%s" % (
270 if attribute.checkpoint_key not in saveable.name:
279 saveable_factory, attribute.checkpoint_key,
293 op=maybe_saveable, name=attribute.checkpoint_key))
297 if attribute.checkpoint_key not in saveable.name:
302 attribute.checkpoint_key))
Dbase.py348 checkpoint_key = serialized_tensor.checkpoint_key
349 dtype = self._checkpoint.dtype_map[checkpoint_key]
362 tensor_names=[checkpoint_key],
382 serialized_tensor.checkpoint_key, None)
413 if serialized_tensor.checkpoint_key not in saveable.name:
430 saveable = saveable_factory(name=serialized_tensor.checkpoint_key)
439 named_saveables[serialized_tensor.checkpoint_key] = saveable
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Drestore_ops.cc70 const std::string& checkpoint_key, DataType dtype, in SingleRestore() argument
83 CreateStringVectorTensorHandle(ctx, checkpoint_key, &names_handle)); in SingleRestore()
Drestore_ops.h34 const std::string& checkpoint_key, DataType dtype,
/external/tensorflow/tensorflow/cc/saved_model/
Dbundle_v2_test.cc43 attr.checkpoint_key()); in RestoreVarsAndVerify()
/external/tensorflow/tensorflow/core/protobuf/
Dtrackable_object_graph.proto33 string checkpoint_key = 3; field
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py398 def save_fn(checkpoint_key): argument
399 maybe_saveable = saveable_factory(name=checkpoint_key)
/external/tensorflow/tensorflow/python/keras/tests/
Dtracking_util_with_v1_optimizers_test.py175 .original_variable_node_id].attributes[0].checkpoint_key)
181 .slot_variable_node_id].attributes[0].checkpoint_key)
Dtracking_util_test.py192 serialized_slot_keys.append(attribute.checkpoint_key)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3211 const auto& checkpoint_key = serialized_tensor_attr->checkpoint_key(); in CreateSavedModelIR() local
3216 saved_model->variable_reader()->Lookup(checkpoint_key, &value), in CreateSavedModelIR()
3218 checkpoint_key); in CreateSavedModelIR()
/external/tensorflow/tensorflow/python/training/
Dsaver.py1638 names_to_keys[attribute.full_name] = attribute.checkpoint_key