Add TF_GraphVersions() to C API and use in Graph.graph_def_versions()

PiperOrigin-RevId: 173902666

1 files changed, 11 insertions, 0 deletions

diff --git a/tensorflow/c/c_api.cc b/tensorflow/c/c_api.cc
index b43d202f4e..6dd1b99910 100644
--- a/tensorflow/c/c_api.cc
+++ b/tensorflow/c/c_api.cc
@@ -81,6 +81,7 @@ using tensorflow::TensorBuffer;
 using tensorflow::TensorId;
 using tensorflow::TensorShape;
 using tensorflow::TensorShapeProto;
+using tensorflow::VersionDef;
 using tensorflow::error::Code;
 using tensorflow::errors::FailedPrecondition;
 using tensorflow::errors::InvalidArgument;
@@ -1809,6 +1810,16 @@ void TF_GraphGetOpDef(TF_Graph* graph, const char* op_name,
   status->status = MessageToBuffer(*op_def, output_op_def);
 }
 
+void TF_GraphVersions(TF_Graph* graph, TF_Buffer* output_version_def,
+                      TF_Status* status) {
+  VersionDef versions;
+  {
+    mutex_lock l(graph->mu);
+    versions = graph->graph.versions();
+  }
+  status->status = MessageToBuffer(versions, output_version_def);
+}
+
 TF_ImportGraphDefOptions* TF_NewImportGraphDefOptions() {
   return new TF_ImportGraphDefOptions;
 }