Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/checkpoint/
Drestore.py95 attributes = self.object_proto.attributes
98 not self.object_proto.children)
116 for serialized_tensor in self.object_proto.attributes:
157 if not (self.object_proto.attributes or recorded_registered_saver):
171 name = self.object_proto.registered_saver.object_name
186 self.object_proto.attributes[0].checkpoint_key: self.trackable
201 for serialized_tensor in self.object_proto.attributes:
213 self.object_proto.attributes[0].checkpoint_key) + suffix
258 for serialized_tensor in self.object_proto.attributes:
350 def object_proto(self): member in CheckpointPosition
[all …]
Dutil.py114 for node_id, object_proto in enumerate(object_graph_proto.nodes):
115 if (object_proto.attributes or object_proto.slot_variables or
116 object_proto.HasField("registered_saver")):
118 for child_proto in object_proto.children:
139 for node_id, object_proto in enumerate(object_graph_proto.nodes):
140 object_proto.has_checkpoint_values.value = bool(
Dsave_util_v1.py125 object_proto = object_graph_proto.nodes[node_ids[trackable]]
126 object_proto.registered_saver.name = saver_name
127 object_proto.registered_saver.object_name = object_name
149 object_proto = object_graph_proto.nodes[node_ids[trackable]]
218 object_proto.attributes.add(
223 object_proto.attributes.add(
239 object_proto = object_graph_proto.nodes.add()
240 object_proto.slot_variables.extend(slot_variables.get(trackable, ()))
242 child_proto = object_proto.children.add()
Dcheckpoint_test.py1157 object_proto = trackable_utils.object_metadata(save_path)
1158 root_attributes = object_proto.nodes[0].attributes
1166 child_attributes = object_proto.nodes[1].attributes
Dcheckpoint.py344 key = position.object_proto.attributes[0].checkpoint_key
/external/tensorflow/tensorflow/python/eager/
Dfunction_saved_model_utils.py75 def _serialize_to_proto(self, object_proto=None, **kwargs): argument
76 object_proto.constant.operation = self._exported_tensor.op.name
79 def _deserialize_from_proto(cls, object_proto, operation_attributes, argument
82 operation_attributes[object_proto.constant.operation]["value"].tensor)
/external/tensorflow/tensorflow/python/trackable/
Dasset.py92 def _deserialize_from_proto(cls, object_proto, export_dir, asset_file_def, argument
94 proto = object_proto.asset
Dbase.py768 def _serialize_to_proto(self, object_proto=None, **kwargs): argument
788 del object_proto, kwargs # Unused.
796 object_proto=None, argument
871 del (proto, dependencies, object_proto, export_dir, asset_file_def,
Dresource.py295 def _deserialize_from_proto(cls, object_proto, dependencies, **unused_kwargs): argument
296 obj = cls(device=object_proto.resource.device)
/external/tensorflow/tensorflow/python/saved_model/
Dsave.py343 object_proto = proto.nodes.add()
344 object_proto.slot_variables.extend(self._slot_variables.get(node, ()))
348 child_proto = object_proto.children.add()
352 child_proto = object_proto.dependencies.add()
362 saveable_object_proto = object_proto.saveable_objects[local_name]
367 object_proto.registered_saver = self._obj_to_registered_saver[node]
1013 …serialized_user_proto = obj._serialize_to_proto(object_proto=proto) # pylint: disable=protected-a…
Dload.py321 for node_id, object_proto in self._iter_all_nodes():
322 self._add_object_graph_edges(object_proto, node_id)
603 object_proto=proto,
/external/tensorflow/tensorflow/dtensor/python/
Dd_checkpoint.py201 key = position.object_proto.attributes[0].checkpoint_key