Searched refs:object_graph_tensor (Results 1 – 3 of 3) sorted by relevance
72 Tensor object_graph_tensor; in ReadCheckpointObjectGraph() local74 bundle_reader->Lookup(kObjectGraphProtoKey, &object_graph_tensor), in ReadCheckpointObjectGraph()76 if (object_graph_tensor.dtype() != DT_STRING || in ReadCheckpointObjectGraph()77 object_graph_tensor.dims() != 0 || in ReadCheckpointObjectGraph()78 object_graph_tensor.NumElements() != 1) { in ReadCheckpointObjectGraph()85 object_graph_tensor.tensor_data().data()); in ReadCheckpointObjectGraph()
1117 def _gather_saveables(self, object_graph_tensor=None): argument1121 if object_graph_tensor is None:1123 object_graph_tensor = constant_op.constant(1127 {object_graph_tensor: graph_proto.SerializeToString()})1131 tensor=object_graph_tensor, name=base.OBJECT_GRAPH_PROTO_KEY))1136 object_graph_tensor, argument1154 object_graph_tensor=object_graph_tensor)1206 object_graph_tensor = self._object_graph_feed_tensor1213 object_graph_tensor = None1217 file_prefix_tensor, object_graph_tensor, options)
426 object_graph_tensor = constant_op.constant(430 tensor=object_graph_tensor,