Searched refs:checkpoint_key (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
D | tf_saved_model_api.cc | 143 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/ |
D | graph_view.py | 262 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))
|
D | base.py | 348 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/ |
D | restore_ops.cc | 70 const std::string& checkpoint_key, DataType dtype, in SingleRestore() argument 83 CreateStringVectorTensorHandle(ctx, checkpoint_key, &names_handle)); in SingleRestore()
|
D | restore_ops.h | 34 const std::string& checkpoint_key, DataType dtype,
|
/external/tensorflow/tensorflow/cc/saved_model/ |
D | bundle_v2_test.cc | 43 attr.checkpoint_key()); in RestoreVarsAndVerify()
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | trackable_object_graph.proto | 33 string checkpoint_key = 3; field
|
/external/tensorflow/tensorflow/python/training/saving/ |
D | saveable_object_util.py | 398 def save_fn(checkpoint_key): argument 399 maybe_saveable = saveable_factory(name=checkpoint_key)
|
/external/tensorflow/tensorflow/python/keras/tests/ |
D | tracking_util_with_v1_optimizers_test.py | 175 .original_variable_node_id].attributes[0].checkpoint_key) 181 .slot_variable_node_id].attributes[0].checkpoint_key)
|
D | tracking_util_test.py | 192 serialized_slot_keys.append(attribute.checkpoint_key)
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
D | import_model.cc | 3211 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/ |
D | saver.py | 1638 names_to_keys[attribute.full_name] = attribute.checkpoint_key
|