Lines Matching refs:_checkpoint
84 """Indicates a position within a `_Checkpoint`."""
90 checkpoint: A _Checkpoint object.
93 self._checkpoint = checkpoint
103 self._checkpoint.restore_ops.extend(restore_ops)
176 dtype = self._checkpoint.dtype_map[checkpoint_key]
180 prefix=self._checkpoint.save_path,
190 return self._checkpoint
194 return self._checkpoint.object_by_proto_id[self._proto_id]
198 return self._checkpoint.object_graph_proto.nodes[self._proto_id]
202 return self._checkpoint.restore_uid
228 class _Checkpoint(object):
