overload Graph.Import(byte[] bytes)

Oceania2018 · Oceania2018 · commit 92b3a88be773 · 2019-06-26T20:47:26.000-05:00
diff --git a/src/TensorFlowNET.Core/Graphs/Graph.Import.cs b/src/TensorFlowNET.Core/Graphs/Graph.Import.cs
@@ -35,6 +35,14 @@ public Status Import(string file_path)
             return Status;
         }
 
+        public Status Import(byte[] bytes)
+        {
+            var graph_def = new Tensorflow.Buffer(bytes);
+            var opts = c_api.TF_NewImportGraphDefOptions();
+            c_api.TF_GraphImportGraphDef(_handle, graph_def, opts, Status);
+            return Status;
+        }
+
         public static Graph ImportFromPB(string file_path)
         {
             var graph = tf.Graph().as_default();

Original file line number	Diff line number	Diff line change
`@@ -35,6 +35,14 @@ public Status Import(string file_path)`
`35`	`35`	`return Status;`
`36`	`36`	`}`
`37`	`37`
	`38`	`+ public Status Import(byte[] bytes)`
	`39`	`+ {`
	`40`	`+ var graph_def = new Tensorflow.Buffer(bytes);`
	`41`	`+ var opts = c_api.TF_NewImportGraphDefOptions();`
	`42`	`+ c_api.TF_GraphImportGraphDef(_handle, graph_def, opts, Status);`
	`43`	`+ return Status;`
	`44`	`+ }`
	`45`	`+`
`38`	`46`	`public static Graph ImportFromPB(string file_path)`
`39`	`47`	`{`
`40`	`48`	`var graph = tf.Graph().as_default();`