graph_io.write_graph

Oceania2018 · Oceania2018 · commit fe06a29c975b · 2019-02-11T06:16:02.000-06:00
diff --git a/src/TensorFlowNET.Core/Graphs/Graph.Export.cs b/src/TensorFlowNET.Core/Graphs/Graph.Export.cs
@@ -17,5 +17,18 @@ public Buffer ToGraphDef(Status s)
 
             return buffer;
         }
+
+        public GraphDef _as_graph_def()
+        {
+            var buffer = ToGraphDef(Status);
+            Status.Check();
+            var def = GraphDef.Parser.ParseFrom(buffer);
+            buffer.Dispose();
+
+            // Strip the experimental library field iff it's empty.
+            // if(def.Library.Function.Count == 0)
+
+            return def;
+        }
     }
 }
diff --git a/src/TensorFlowNET.Core/Graphs/Graph.cs b/src/TensorFlowNET.Core/Graphs/Graph.cs
@@ -53,7 +53,7 @@ private Tensor _as_graph_element(object obj)
 
             return null;
         }
-
+       
         private ITensorOrOperation _as_graph_element_locked(object obj, bool allow_tensor = true, bool allow_operation = true)
         {
             string types_str = "";
diff --git a/src/TensorFlowNET.Core/Graphs/graph_io.py.cs b/src/TensorFlowNET.Core/Graphs/graph_io.py.cs
@@ -0,0 +1,21 @@
+﻿using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Text;
+
+namespace Tensorflow
+{
+    public class graph_io
+    {
+        public static string write_graph(Graph graph, string logdir, string name, bool as_text = true)
+        {
+            var def = graph._as_graph_def();
+            string path = Path.Combine(logdir, name);
+            string text = def.ToString();
+            if (as_text)
+                File.WriteAllText(path, text);
+
+            return path;
+        }
+    }
+}
diff --git a/src/TensorFlowNET.Core/Train/tf.optimizers.cs b/src/TensorFlowNET.Core/Train/tf.optimizers.cs
@@ -1,5 +1,6 @@
 ﻿using System;
 using System.Collections.Generic;
+using System.IO;
 using System.Text;
 
 namespace Tensorflow
@@ -11,6 +12,8 @@ public static class train
             public static Optimizer GradientDescentOptimizer(double learning_rate) => new GradientDescentOptimizer(learning_rate);
 
             public static Saver Saver() => new Saver();
+
+            public static string write_graph(Graph graph, string logdir, string name, bool as_text = true) => graph_io.write_graph(graph, logdir, name, as_text);
         }
     }
 }
diff --git a/test/TensorFlowNET.UnitTest/TrainSaverTest.cs b/test/TensorFlowNET.UnitTest/TrainSaverTest.cs
@@ -9,6 +9,14 @@ namespace TensorFlowNET.UnitTest
     [TestClass]
     public class TrainSaverTest : Python
     {
+        [TestMethod]
+        public void WriteGraph()
+        {
+            var v = tf.Variable(0, name: "my_variable");
+            var sess = tf.Session();
+            tf.train.write_graph(sess.graph, "/tmp/my-model", "train.pbtxt");
+        }
+
         [TestMethod]
         public void Save()
         {

Original file line number	Diff line number	Diff line change
`@@ -17,5 +17,18 @@ public Buffer ToGraphDef(Status s)`
`17`	`17`
`18`	`18`	`return buffer;`
`19`	`19`	`}`
	`20`	`+`
	`21`	`+ public GraphDef _as_graph_def()`
	`22`	`+ {`
	`23`	`+ var buffer = ToGraphDef(Status);`
	`24`	`+ Status.Check();`
	`25`	`+ var def = GraphDef.Parser.ParseFrom(buffer);`
	`26`	`+ buffer.Dispose();`
	`27`	`+`
	`28`	`+ // Strip the experimental library field iff it's empty.`
	`29`	`+ // if(def.Library.Function.Count == 0)`
	`30`	`+`
	`31`	`+ return def;`
	`32`	`+ }`
`20`	`33`	`}`
`21`	`34`	`}`
Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,7 @@ private Tensor _as_graph_element(object obj)`
`53`	`53`
`54`	`54`	`return null;`
`55`	`55`	`}`
`56`		`-`
	`56`	`+`
`57`	`57`	`private ITensorOrOperation _as_graph_element_locked(object obj, bool allow_tensor = true, bool allow_operation = true)`
`58`	`58`	`{`
`59`	`59`	`string types_str = "";`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,6 @@`
`1`	`1`	`using System;`
`2`	`2`	`using System.Collections.Generic;`
	`3`	`+using System.IO;`
`3`	`4`	`using System.Text;`
`4`	`5`
`5`	`6`	`namespace Tensorflow`
`@@ -11,6 +12,8 @@ public static class train`
`11`	`12`	`public static Optimizer GradientDescentOptimizer(double learning_rate) => new GradientDescentOptimizer(learning_rate);`
`12`	`13`
`13`	`14`	`public static Saver Saver() => new Saver();`
	`15`	`+`
	`16`	`+ public static string write_graph(Graph graph, string logdir, string name, bool as_text = true) => graph_io.write_graph(graph, logdir, name, as_text);`
`14`	`17`	`}`
`15`	`18`	`}`
`16`	`19`	`}`