forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSession.cs
More file actions
77 lines (59 loc) · 1.76 KB
/
Session.cs
File metadata and controls
77 lines (59 loc) · 1.76 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
using System;
using System.Collections.Generic;
using System.Text;
namespace Tensorflow
{
public class Session : BaseSession, IPython
{
public Session(string target = "", Graph g = null)
: base(target, g, null)
{
}
public Session(IntPtr handle)
: base("", null, null)
{
_session = handle;
}
public Session(Graph g, SessionOptions opts = null, Status s = null)
: base("", g, opts)
{
if (s == null)
s = Status;
}
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._session;
public static implicit operator Session(IntPtr handle) => new Session(handle);
public void close()
{
Dispose();
}
public void Dispose()
{
c_api.TF_DeleteSession(_session, Status);
Status.Dispose();
}
public void __enter__()
{
}
public void __exit__()
{
}
}
}