Searched refs:object_graph_tensor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/ |
| H A D | bundle_v2.cc | 75 Tensor object_graph_tensor; in ReadCheckpointObjectGraph() local
77 bundle_reader->Lookup(kObjectGraphProtoKey, &object_graph_tensor), in ReadCheckpointObjectGraph()
79 if (object_graph_tensor.dtype() != DT_STRING || in ReadCheckpointObjectGraph()
80 object_graph_tensor.dims() != 0 || in ReadCheckpointObjectGraph()
81 object_graph_tensor.NumElements() != 1) { in ReadCheckpointObjectGraph()
88 object_graph_tensor.tensor_data().data()); in ReadCheckpointObjectGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libexport/ |
| H A D | load.cc | 67 tensorflow::Tensor object_graph_tensor; in Load() local
69 tensorflow::kObjectGraphProtoKey, &object_graph_tensor)); in Load()
72 object_graph_tensor.tensor_data().data()); in Load()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | checkpoint.py | 1180 def _gather_saveables(self, object_graph_tensor=None): argument
1188 if object_graph_tensor is None:
1190 object_graph_tensor = constant_op.constant(
1194 {object_graph_tensor: graph_proto.SerializeToString()})
1198 tensor=object_graph_tensor, name=base.OBJECT_GRAPH_PROTO_KEY))
1204 object_graph_tensor, argument
1226 object_graph_tensor=object_graph_tensor)
1336 object_graph_tensor = self._object_graph_feed_tensor
1343 object_graph_tensor = None
1349 file_prefix_tensor, object_graph_tensor, options, write_done_callback)
|
| H A D | save_util_v1.py | 305 object_graph_tensor = constant_op.constant(
309 tensor=object_graph_tensor, name=base.OBJECT_GRAPH_PROTO_KEY))
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/ |
| H A D | d_checkpoint.py | 234 object_graph_tensor, argument
257 object_graph_tensor=object_graph_tensor)
|