change Session(ConfigProto).

Author: Oceania2018

src/TensorFlowNET.Core/Sessions/BaseSession.cs

@@ -36,19 +36,20 @@ public class BaseSession : DisposableObject
         protected byte[] _target;
         public Graph graph => _graph;
 
-        public BaseSession(string target = "", Graph g = null, SessionOptions opts = null, Status status = null)
+        public BaseSession(string target = "", Graph g = null, ConfigProto config = null, Status status = null)
         {
             _graph = g ?? ops.get_default_graph();
             _graph.as_default();
             _target = Encoding.UTF8.GetBytes(target);
 
-            SessionOptions lopts = opts ?? new SessionOptions();
-
-            lock (Locks.ProcessWide)
+            using (var opts = new SessionOptions(target, config))
             {
-                status = status ?? new Status();
-                _handle = c_api.TF_NewSession(_graph, opts ?? lopts, status);
-                status.Check(true);
+                lock (Locks.ProcessWide)
+                {
+                    status = status ?? new Status();
+                    _handle = c_api.TF_NewSession(_graph, opts, status);
+                    status.Check(true);
+                }
             }
         }
 

src/TensorFlowNET.Core/Sessions/Session.cs

@@ -32,7 +32,7 @@ public Session(IntPtr handle, Graph g = null) : base("", g, null)
             _handle = handle;
         }
 
-        public Session(Graph g, SessionOptions opts = null, Status s = null) : base("", g, opts, s)
+        public Session(Graph g, ConfigProto config = null, Status s = null) : base("", g, config, s)
         { }
 
         public Session as_default()

src/TensorFlowNET.Core/Sessions/SessionOptions.cs

@@ -20,11 +20,14 @@ limitations under the License.
 
 namespace Tensorflow
 {
-    public class SessionOptions : DisposableObject
+    internal class SessionOptions : DisposableObject
     {
-        public SessionOptions()
+        public SessionOptions(string target = "", ConfigProto config = null)
         {
             _handle = c_api.TF_NewSessionOptions();
+            c_api.TF_SetTarget(_handle, target);
+            if (config != null)
+                SetConfig(config);
         }
 
         public SessionOptions(IntPtr handle)
@@ -35,10 +38,10 @@ public SessionOptions(IntPtr handle)
         protected override void DisposeUnmanagedResources(IntPtr handle)
             => c_api.TF_DeleteSessionOptions(handle);
 
-        public void SetConfig(ConfigProto config)
+        private void SetConfig(ConfigProto config)
         {
-            var bytes = config.ToByteArray(); //TODO! we can use WriteTo
-            var proto = Marshal.AllocHGlobal(bytes.Length); //TODO! potential memory leak
+            var bytes = config.ToByteArray();
+            var proto = Marshal.AllocHGlobal(bytes.Length);
             Marshal.Copy(bytes, 0, proto, bytes.Length);
 
             using (var status = new Status())

src/TensorFlowNET.Core/Sessions/TF_DeprecatedSession.cs

@@ -116,6 +116,9 @@ public static extern unsafe void TF_SessionRun(IntPtr session, TF_Buffer* run_op
         /// <param name="proto_len">size_t</param>
         /// <param name="status">TF_Status*</param>
         [DllImport(TensorFlowLibName)]
-        public static extern unsafe void TF_SetConfig(IntPtr options, IntPtr proto, ulong proto_len, IntPtr status);
+        public static extern void TF_SetConfig(IntPtr options, IntPtr proto, ulong proto_len, IntPtr status);
+
+        [DllImport(TensorFlowLibName)]
+        public static extern void TF_SetTarget(IntPtr options, string target);
     }
 }