Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/checkpoint/
Drestore.py117 checkpoint_key = serialized_tensor.checkpoint_key
118 dtype = self._checkpoint.dtype_map[checkpoint_key]
131 tensor_names=[checkpoint_key],
186 self.object_proto.attributes[0].checkpoint_key: self.trackable
213 self.object_proto.attributes[0].checkpoint_key) + suffix
263 serialized_tensor.checkpoint_key, None)
298 if serialized_tensor.checkpoint_key not in saveable.name:
318 named_saveables[serialized_tensor.checkpoint_key] = saveable
372 return self._checkpoint.shape_map[serialized_tensor.checkpoint_key]
562 def _extract_saveable_name(checkpoint_key): argument
[all …]
Dfunctional_saver.py63 for checkpoint_key, tensor_slices in self._tensor_slice_dict.items():
75 tensor_names.append(checkpoint_key)
98 for checkpoint_key, tensor_slices in self._tensor_slice_dict.items():
106 tensor_names.append(checkpoint_key)
114 for checkpoint_key, tensor_slices in self._tensor_slice_dict.items():
117 restored_tensor_dict.setdefault(checkpoint_key, {})[slice_spec] = (
217 for checkpoint_key, maybe_tensor in tensor_dict.items():
223 if (checkpoint_key, slice_spec) in self._keys_to_restore_fn:
229 self._keys_to_restore_fn[(checkpoint_key, slice_spec)] = restore_fn
231 (checkpoint_key, slice_spec))
[all …]
Dsave_util_v1.py80 checkpoint_key = trackable_utils.checkpoint_key(object_name, key_suffix)
91 checkpoint_key=checkpoint_key))
158 key = factory_data.checkpoint_key
220 checkpoint_key=local_key,
225 checkpoint_key=key,
Dsaveable_compat_test.py58 table_proto.attributes[0].checkpoint_key],
60 legacy_table_proto.attributes[0].checkpoint_key])
Dcheckpoint_test.py1162 {attr.name: attr.checkpoint_key for attr in root_attributes})
1170 {attr.name: attr.checkpoint_key for attr in child_attributes})
Dcheckpoint.py344 key = position.object_proto.attributes[0].checkpoint_key
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dtf_saved_model_api.cc144 const std::string& checkpoint_key = attribute->checkpoint_key(); in RestoreCheckpoint() local
145 if (!bundle->variable_reader()->Contains(checkpoint_key)) { in RestoreCheckpoint()
146 LOG(WARNING) << "No checkpoint entry found for " << checkpoint_key in RestoreCheckpoint()
156 context, variables_path_prefix, checkpoint_key, variable->dtype(), in RestoreCheckpoint()
/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/python/trackable/
Dtrackable_utils.py136 def checkpoint_key(object_path, local_name): function
/external/tensorflow/tensorflow/cc/saved_model/
Dbundle_v2_test.cc44 attr.checkpoint_key()); in RestoreVarsAndVerify()
/external/tensorflow/tensorflow/cc/experimental/libexport/
Dload.cc107 return serialized_tensor->checkpoint_key(); in GetVariableCheckpointKey()
/external/tensorflow/tensorflow/core/protobuf/
Dtrackable_object_graph.proto35 string checkpoint_key = 3; field
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py415 def save_fn(checkpoint_key): argument
416 maybe_saveable = saveable_factory(name=checkpoint_key)
/external/tensorflow/tensorflow/dtensor/python/
Dd_checkpoint.py201 key = position.object_proto.attributes[0].checkpoint_key
/external/tensorflow/tensorflow/python/training/
Dsaver.py1754 names_to_keys[attribute.full_name] = attribute.checkpoint_key
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3454 const auto& checkpoint_key = serialized_tensor_attr->checkpoint_key(); in CreateSavedModelIR() local
3459 saved_model->variable_reader()->Lookup(checkpoint_key, &value), in CreateSavedModelIR()
3461 checkpoint_key); in CreateSavedModelIR()