TensorFlow.NET/src/TensorFlowNET.Core/Sessions/Session.cs

using System;
using System.Collections.Generic;
using System.Text;

namespace Tensorflow
{
    public class Session : BaseSession, IPython
    {
        private IntPtr _handle;
        public Status Status  = new Status();
        public SessionOptions Options { get; }
        public Graph graph;

        public Session(string target = "", Graph graph = null):base(target,graph,null)
        {
            if(graph == null)
            {
                graph = tf.get_default_graph();
            }
            this.graph = graph;
            Options = new SessionOptions();
            _handle = c_api.TF_NewSession(graph, Options, Status);
            //why create session again. already created session in BaseSession.
            Status.Check();
        }

        public Session(IntPtr handle):base("",null,null)
        {
            _handle = handle;
        }

        public Session(Graph g, SessionOptions opts = null, Status s = null)
            :base(string.Empty,g,opts)
        {
            s = s ?? Status;           
            graph = g;
            Options = opts ?? new SessionOptions();
            _handle = c_api.TF_NewSession(graph, Options, s);
            Status.Check(true);
        }

        public Session as_default()
        {
            tf.defaultSession = this;
            return this;
        }

        public static Session LoadFromSavedModel(string path)
        {
            var graph = c_api.TF_NewGraph();
            var status = new Status();
            var opt = c_api.TF_NewSessionOptions();

            var buffer = new TF_Buffer();
            var sess = c_api.TF_LoadSessionFromSavedModel(opt, IntPtr.Zero, path, new string[0], 0, graph, ref buffer, status);

            //var bytes = new Buffer(buffer.data).Data;
            //var meta_graph = MetaGraphDef.Parser.ParseFrom(bytes);

            status.Check();

            new Graph(graph).as_default();

            return sess;
        }

        public static implicit operator IntPtr(Session session) => session._handle;
        public static implicit operator Session(IntPtr handle) => new Session(handle);

        public void close()
        {
            Dispose();
        }

        public void Dispose()
        {
            Options.Dispose();
            c_api.TF_DeleteSession(_handle, Status);
            Status.Dispose();
        }

        public void __enter__()
        {
            
        }

        public void __exit__()
        {
            
        }
    }
}