Skip to content
Navigation Menu
{{ message }}
forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathc_api.graph.cs
More file actions
251 lines (227 loc) · 12 KB
/
Copy pathc_api.graph.cs
File metadata and controls
251 lines (227 loc) · 12 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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
using System;
using System.Collections.Generic;
using System.Runtime.InteropServices;
using System.Text;
namespace Tensorflow
{
public partial class c_api
{
/// <summary>
/// Destroy an options object. Graph will be deleted once no more
/// TFSession's are referencing it.
/// </summary>
/// <param name="graph"></param>
[DllImport(TensorFlowLibName)]
public static extern void TF_DeleteGraph(IntPtr graph);
/// <summary>
///
/// </summary>
/// <param name="opts">TF_ImportGraphDefOptions*</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_DeleteImportGraphDefOptions(IntPtr opts);
/// <summary>
/// Deletes a results object returned by TF_GraphImportGraphDefWithResults().
/// </summary>
/// <param name="results"></param>
[DllImport(TensorFlowLibName)]
public static extern void TF_DeleteImportGraphDefResults(IntPtr results);
[DllImport(TensorFlowLibName)]
public static extern void TF_GraphGetOpDef(IntPtr graph, string op_name, IntPtr output_op_def, IntPtr status);
/// <summary>
/// Returns the shape of the Tensor referenced by `output` in `graph`
/// into `dims`. `dims` must be an array large enough to hold `num_dims`
/// entries (e.g., the return value of TF_GraphGetTensorNumDims).
/// </summary>
/// <param name="graph"></param>
/// <param name="output"></param>
/// <param name="dims"></param>
/// <param name="num_dims"></param>
/// <param name="status"></param>
[DllImport(TensorFlowLibName)]
public static extern void TF_GraphGetTensorShape(IntPtr graph, TF_Output output, long[] dims, int num_dims, IntPtr status);
/// <summary>
/// Import the graph serialized in `graph_def` into `graph`.
/// Convenience function for when only return outputs are needed.
///
/// `num_return_outputs` must be the number of return outputs added (i.e. the
/// result of TF_ImportGraphDefOptionsNumReturnOutputs()). If
/// `num_return_outputs` is non-zero, `return_outputs` must be of length
/// `num_return_outputs`. Otherwise it can be null.
/// </summary>
/// <param name="graph">TF_Graph* graph</param>
/// <param name="graph_def">const TF_Buffer*</param>
/// <param name="options">const TF_ImportGraphDefOptions*</param>
/// <param name="return_outputs">TF_Output*</param>
/// <param name="num_return_outputs">int</param>
/// <param name="status">TF_Status*</param>
[DllImport(TensorFlowLibName)]
public static extern unsafe void TF_GraphImportGraphDefWithReturnOutputs(IntPtr graph, IntPtr graph_def, IntPtr options, IntPtr return_outputs, int num_return_outputs, IntPtr status);
/// <summary>
/// Import the graph serialized in `graph_def` into `graph`. Returns nullptr and
/// a bad status on error. Otherwise, returns a populated
/// TF_ImportGraphDefResults instance. The returned instance must be deleted via
/// TF_DeleteImportGraphDefResults().
/// </summary>
/// <param name="graph">TF_Graph*</param>
/// <param name="graph_def">const TF_Buffer*</param>
/// <param name="options">const TF_ImportGraphDefOptions*</param>
/// <param name="status">TF_Status*</param>
/// <returns>TF_ImportGraphDefResults*</returns>
[DllImport(TensorFlowLibName)]
public static extern IntPtr TF_GraphImportGraphDefWithResults(IntPtr graph, IntPtr graph_def, IntPtr options, IntPtr status);
/// <summary>
/// Import the graph serialized in `graph_def` into `graph`.
/// </summary>
/// <param name="graph">TF_Graph*</param>
/// <param name="graph_def">TF_Buffer*</param>
/// <param name="options">TF_ImportGraphDefOptions*</param>
/// <param name="status">TF_Status*</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_GraphImportGraphDef(IntPtr graph, IntPtr graph_def, IntPtr options, IntPtr status);
/// <summary>
/// Iterate through the operations of a graph.
/// </summary>
/// <param name="graph"></param>
/// <param name="pos"></param>
/// <returns></returns>
[DllImport(TensorFlowLibName)]
public static extern IntPtr TF_GraphNextOperation(IntPtr graph, ref uint pos);
/// <summary>
/// Returns the operation in the graph with `oper_name`. Returns nullptr if
/// no operation found.
/// </summary>
/// <param name="graph"></param>
/// <param name="oper_name"></param>
/// <returns></returns>
[DllImport(TensorFlowLibName)]
public static extern IntPtr TF_GraphOperationByName(IntPtr graph, string oper_name);
/// <summary>
/// Sets the shape of the Tensor referenced by `output` in `graph` to
/// the shape described by `dims` and `num_dims`.
/// </summary>
[DllImport(TensorFlowLibName)]
public static extern void TF_GraphSetTensorShape(IntPtr graph, TF_Output output, long[] dims, int num_dims, IntPtr status);
/// <summary>
/// Write out a serialized representation of `graph` (as a GraphDef protocol
/// message) to `output_graph_def` (allocated by TF_NewBuffer()).
/// </summary>
/// <param name="graph">TF_Graph*</param>
/// <param name="output_graph_def">TF_Buffer*</param>
/// <param name="status">TF_Status*</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_GraphToGraphDef(IntPtr graph, IntPtr output_graph_def, IntPtr status);
/// <summary>
/// Returns the number of dimensions of the Tensor referenced by `output`
/// in `graph`.
///
/// If the number of dimensions in the shape is unknown, returns -1.
/// </summary>
/// <param name="graph"></param>
/// <param name="output"></param>
/// <param name="status"></param>
/// <returns></returns>
[DllImport(TensorFlowLibName)]
public static extern int TF_GraphGetTensorNumDims(IntPtr graph, TF_Output output, IntPtr status);
/// <summary>
/// Cause the imported graph to have a control dependency on `oper`. `oper`
/// should exist in the graph being imported into.
/// </summary>
/// <param name="opts"></param>
/// <param name="oper"></param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefOptionsAddControlDependency(IntPtr opts, IntPtr oper);
/// <summary>
/// Set any imported nodes with input `src_name:src_index` to have that input
/// replaced with `dst`. `src_name` refers to a node in the graph to be imported,
/// `dst` references a node already existing in the graph being imported into.
/// `src_name` is copied and has no lifetime requirements.
/// </summary>
/// <param name="opts">TF_ImportGraphDefOptions*</param>
/// <param name="src_name">const char*</param>
/// <param name="src_index">int</param>
/// <param name="dst">TF_Output</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefOptionsAddInputMapping(IntPtr opts, string src_name, int src_index, TF_Output dst);
/// <summary>
/// Add an operation in `graph_def` to be returned via the `return_opers` output
/// parameter of TF_GraphImportGraphDef(). `oper_name` is copied and has no
// lifetime requirements.
/// </summary>
/// <param name="opts">TF_ImportGraphDefOptions* opts</param>
/// <param name="oper_name">const char*</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefOptionsAddReturnOperation(IntPtr opts, string oper_name);
/// <summary>
/// Add an output in `graph_def` to be returned via the `return_outputs` output
/// parameter of TF_GraphImportGraphDef(). If the output is remapped via an input
/// mapping, the corresponding existing tensor in `graph` will be returned.
/// `oper_name` is copied and has no lifetime requirements.
/// </summary>
/// <param name="opts">TF_ImportGraphDefOptions*</param>
/// <param name="oper_name">const char*</param>
/// <param name="index">int</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefOptionsAddReturnOutput(IntPtr opts, string oper_name, int index);
/// <summary>
/// Returns the number of return operations added via
/// TF_ImportGraphDefOptionsAddReturnOperation().
/// </summary>
/// <param name="opts"></param>
/// <returns></returns>
[DllImport(TensorFlowLibName)]
public static extern int TF_ImportGraphDefOptionsNumReturnOperations(IntPtr opts);
/// <summary>
/// Returns the number of return outputs added via
/// TF_ImportGraphDefOptionsAddReturnOutput().
/// </summary>
/// <param name="opts">const TF_ImportGraphDefOptions*</param>
/// <returns></returns>
[DllImport(TensorFlowLibName)]
public static extern int TF_ImportGraphDefOptionsNumReturnOutputs(IntPtr opts);
/// <summary>
/// Set any imported nodes with control input `src_name` to have that input
/// replaced with `dst`. `src_name` refers to a node in the graph to be imported,
/// `dst` references an operation already existing in the graph being imported
/// into. `src_name` is copied and has no lifetime requirements.
/// </summary>
/// <param name="opts">TF_ImportGraphDefOptions*</param>
/// <param name="src_name">const char*</param>
/// <param name="dst">TF_Operation*</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefOptionsRemapControlDependency(IntPtr opts, string src_name, IntPtr dst);
/// <summary>
/// Set the prefix to be prepended to the names of nodes in `graph_def` that will
/// be imported into `graph`. `prefix` is copied and has no lifetime
/// requirements.
/// </summary>
/// <param name="ops"></param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefOptionsSetPrefix(IntPtr ops, string prefix);
/// <summary>
/// Fetches the return operations requested via
/// TF_ImportGraphDefOptionsAddReturnOperation(). The number of fetched
/// operations is returned in `num_opers`. The array of return operations is
/// returned in `opers`. `*opers` is owned by and has the lifetime of `results`.
/// </summary>
/// <param name="results">TF_ImportGraphDefResults*</param>
/// <param name="num_opers">int*</param>
/// <param name="opers">TF_Operation***</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefResultsReturnOperations(IntPtr results, ref int num_opers, ref TF_Operation opers);
/// <summary>
/// Fetches the return outputs requested via
/// TF_ImportGraphDefOptionsAddReturnOutput(). The number of fetched outputs is
/// returned in `num_outputs`. The array of return outputs is returned in
/// `outputs`. `*outputs` is owned by and has the lifetime of `results`.
/// </summary>
/// <param name="results">TF_ImportGraphDefResults* results</param>
/// <param name="num_outputs">int*</param>
/// <param name="outputs">TF_Output**</param>
[DllImport(TensorFlowLibName)]
public static extern void TF_ImportGraphDefResultsReturnOutputs(IntPtr results, ref int num_outputs, ref IntPtr outputs);
[DllImport(TensorFlowLibName)]
public static extern IntPtr TF_NewGraph();
[DllImport(TensorFlowLibName)]
public static extern IntPtr TF_NewImportGraphDefOptions();
}
}
You can’t perform that action at this time.