rknncli 0.2.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- rknncli/__init__.py +3 -0
- rknncli/cli.py +229 -0
- rknncli/parser.py +320 -0
- rknncli/schema/rknn/Graph.py +371 -0
- rknncli/schema/rknn/Model.py +422 -0
- rknncli/schema/rknn/Node.py +380 -0
- rknncli/schema/rknn/Tensor.py +697 -0
- rknncli/schema/rknn/Type1.py +94 -0
- rknncli/schema/rknn/Type2.py +255 -0
- rknncli/schema/rknn/Type3.py +94 -0
- rknncli/schema/rknn/__init__.py +0 -0
- rknncli-0.2.0.dist-info/METADATA +69 -0
- rknncli-0.2.0.dist-info/RECORD +16 -0
- rknncli-0.2.0.dist-info/WHEEL +5 -0
- rknncli-0.2.0.dist-info/entry_points.txt +2 -0
- rknncli-0.2.0.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,697 @@
|
|
|
1
|
+
# automatically generated by the FlatBuffers compiler, do not modify
|
|
2
|
+
|
|
3
|
+
# namespace: rknn
|
|
4
|
+
|
|
5
|
+
import flatbuffers
|
|
6
|
+
from flatbuffers.compat import import_numpy
|
|
7
|
+
np = import_numpy()
|
|
8
|
+
|
|
9
|
+
class Tensor(object):
|
|
10
|
+
__slots__ = ['_tab']
|
|
11
|
+
|
|
12
|
+
@classmethod
|
|
13
|
+
def GetRootAs(cls, buf, offset=0):
|
|
14
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
|
|
15
|
+
x = Tensor()
|
|
16
|
+
x.Init(buf, n + offset)
|
|
17
|
+
return x
|
|
18
|
+
|
|
19
|
+
@classmethod
|
|
20
|
+
def GetRootAsTensor(cls, buf, offset=0):
|
|
21
|
+
"""This method is deprecated. Please switch to GetRootAs."""
|
|
22
|
+
return cls.GetRootAs(buf, offset)
|
|
23
|
+
@classmethod
|
|
24
|
+
def TensorBufferHasIdentifier(cls, buf, offset, size_prefixed=False):
|
|
25
|
+
return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x52\x4B\x4E\x4E", size_prefixed=size_prefixed)
|
|
26
|
+
|
|
27
|
+
# Tensor
|
|
28
|
+
def Init(self, buf, pos):
|
|
29
|
+
self._tab = flatbuffers.table.Table(buf, pos)
|
|
30
|
+
|
|
31
|
+
# Tensor
|
|
32
|
+
def DataType(self):
|
|
33
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
|
|
34
|
+
if o != 0:
|
|
35
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
36
|
+
return 0
|
|
37
|
+
|
|
38
|
+
# Tensor
|
|
39
|
+
def Var02(self):
|
|
40
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
|
|
41
|
+
if o != 0:
|
|
42
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
43
|
+
return 0
|
|
44
|
+
|
|
45
|
+
# Tensor
|
|
46
|
+
def Kind(self):
|
|
47
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
|
|
48
|
+
if o != 0:
|
|
49
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos)
|
|
50
|
+
return 0
|
|
51
|
+
|
|
52
|
+
# Tensor
|
|
53
|
+
def Var04(self, j):
|
|
54
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
55
|
+
if o != 0:
|
|
56
|
+
a = self._tab.Vector(o)
|
|
57
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
58
|
+
return 0
|
|
59
|
+
|
|
60
|
+
# Tensor
|
|
61
|
+
def Var04AsNumpy(self):
|
|
62
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
63
|
+
if o != 0:
|
|
64
|
+
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
65
|
+
return 0
|
|
66
|
+
|
|
67
|
+
# Tensor
|
|
68
|
+
def Var04Length(self):
|
|
69
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
70
|
+
if o != 0:
|
|
71
|
+
return self._tab.VectorLen(o)
|
|
72
|
+
return 0
|
|
73
|
+
|
|
74
|
+
# Tensor
|
|
75
|
+
def Var04IsNone(self):
|
|
76
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(10))
|
|
77
|
+
return o == 0
|
|
78
|
+
|
|
79
|
+
# Tensor
|
|
80
|
+
def Shape(self, j):
|
|
81
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
82
|
+
if o != 0:
|
|
83
|
+
a = self._tab.Vector(o)
|
|
84
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
|
|
85
|
+
return 0
|
|
86
|
+
|
|
87
|
+
# Tensor
|
|
88
|
+
def ShapeAsNumpy(self):
|
|
89
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
90
|
+
if o != 0:
|
|
91
|
+
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
|
|
92
|
+
return 0
|
|
93
|
+
|
|
94
|
+
# Tensor
|
|
95
|
+
def ShapeLength(self):
|
|
96
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
97
|
+
if o != 0:
|
|
98
|
+
return self._tab.VectorLen(o)
|
|
99
|
+
return 0
|
|
100
|
+
|
|
101
|
+
# Tensor
|
|
102
|
+
def ShapeIsNone(self):
|
|
103
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12))
|
|
104
|
+
return o == 0
|
|
105
|
+
|
|
106
|
+
# Tensor
|
|
107
|
+
def Name(self):
|
|
108
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
|
|
109
|
+
if o != 0:
|
|
110
|
+
return self._tab.String(o + self._tab.Pos)
|
|
111
|
+
return None
|
|
112
|
+
|
|
113
|
+
# Tensor
|
|
114
|
+
def Var06(self, j):
|
|
115
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
116
|
+
if o != 0:
|
|
117
|
+
a = self._tab.Vector(o)
|
|
118
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
119
|
+
return 0
|
|
120
|
+
|
|
121
|
+
# Tensor
|
|
122
|
+
def Var06AsNumpy(self):
|
|
123
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
124
|
+
if o != 0:
|
|
125
|
+
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int8Flags, o)
|
|
126
|
+
return 0
|
|
127
|
+
|
|
128
|
+
# Tensor
|
|
129
|
+
def Var06Length(self):
|
|
130
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
131
|
+
if o != 0:
|
|
132
|
+
return self._tab.VectorLen(o)
|
|
133
|
+
return 0
|
|
134
|
+
|
|
135
|
+
# Tensor
|
|
136
|
+
def Var06IsNone(self):
|
|
137
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(16))
|
|
138
|
+
return o == 0
|
|
139
|
+
|
|
140
|
+
# Tensor
|
|
141
|
+
def Var07(self):
|
|
142
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(18))
|
|
143
|
+
if o != 0:
|
|
144
|
+
return self._tab.String(o + self._tab.Pos)
|
|
145
|
+
return None
|
|
146
|
+
|
|
147
|
+
# Tensor
|
|
148
|
+
def Var08(self, j):
|
|
149
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
150
|
+
if o != 0:
|
|
151
|
+
a = self._tab.Vector(o)
|
|
152
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
153
|
+
return 0
|
|
154
|
+
|
|
155
|
+
# Tensor
|
|
156
|
+
def Var08AsNumpy(self):
|
|
157
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
158
|
+
if o != 0:
|
|
159
|
+
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int8Flags, o)
|
|
160
|
+
return 0
|
|
161
|
+
|
|
162
|
+
# Tensor
|
|
163
|
+
def Var08Length(self):
|
|
164
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
165
|
+
if o != 0:
|
|
166
|
+
return self._tab.VectorLen(o)
|
|
167
|
+
return 0
|
|
168
|
+
|
|
169
|
+
# Tensor
|
|
170
|
+
def Var08IsNone(self):
|
|
171
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(20))
|
|
172
|
+
return o == 0
|
|
173
|
+
|
|
174
|
+
# Tensor
|
|
175
|
+
def Var09(self, j):
|
|
176
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
177
|
+
if o != 0:
|
|
178
|
+
a = self._tab.Vector(o)
|
|
179
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
180
|
+
return 0
|
|
181
|
+
|
|
182
|
+
# Tensor
|
|
183
|
+
def Var09AsNumpy(self):
|
|
184
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
185
|
+
if o != 0:
|
|
186
|
+
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int8Flags, o)
|
|
187
|
+
return 0
|
|
188
|
+
|
|
189
|
+
# Tensor
|
|
190
|
+
def Var09Length(self):
|
|
191
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
192
|
+
if o != 0:
|
|
193
|
+
return self._tab.VectorLen(o)
|
|
194
|
+
return 0
|
|
195
|
+
|
|
196
|
+
# Tensor
|
|
197
|
+
def Var09IsNone(self):
|
|
198
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
|
|
199
|
+
return o == 0
|
|
200
|
+
|
|
201
|
+
# Tensor
|
|
202
|
+
def Var10(self, j):
|
|
203
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24))
|
|
204
|
+
if o != 0:
|
|
205
|
+
a = self._tab.Vector(o)
|
|
206
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
207
|
+
return 0
|
|
208
|
+
|
|
209
|
+
# Tensor
|
|
210
|
+
def Var10AsNumpy(self):
|
|
211
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24))
|
|
212
|
+
if o != 0:
|
|
213
|
+
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int8Flags, o)
|
|
214
|
+
return 0
|
|
215
|
+
|
|
216
|
+
# Tensor
|
|
217
|
+
def Var10Length(self):
|
|
218
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24))
|
|
219
|
+
if o != 0:
|
|
220
|
+
return self._tab.VectorLen(o)
|
|
221
|
+
return 0
|
|
222
|
+
|
|
223
|
+
# Tensor
|
|
224
|
+
def Var10IsNone(self):
|
|
225
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(24))
|
|
226
|
+
return o == 0
|
|
227
|
+
|
|
228
|
+
# Tensor
|
|
229
|
+
def Var11(self, j):
|
|
230
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(26))
|
|
231
|
+
if o != 0:
|
|
232
|
+
a = self._tab.Vector(o)
|
|
233
|
+
return self._tab.Get(flatbuffers.number_types.Int8Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 1))
|
|
234
|
+
return 0
|
|
235
|
+
|
|
236
|
+
# Tensor
|
|
237
|
+
def Var11AsNumpy(self):
|
|
238
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(26))
|
|
239
|
+
if o != 0:
|
|
240
|
+
return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int8Flags, o)
|
|
241
|
+
return 0
|
|
242
|
+
|
|
243
|
+
# Tensor
|
|
244
|
+
def Var11Length(self):
|
|
245
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(26))
|
|
246
|
+
if o != 0:
|
|
247
|
+
return self._tab.VectorLen(o)
|
|
248
|
+
return 0
|
|
249
|
+
|
|
250
|
+
# Tensor
|
|
251
|
+
def Var11IsNone(self):
|
|
252
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(26))
|
|
253
|
+
return o == 0
|
|
254
|
+
|
|
255
|
+
# Tensor
|
|
256
|
+
def Size(self):
|
|
257
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(28))
|
|
258
|
+
if o != 0:
|
|
259
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
260
|
+
return 0
|
|
261
|
+
|
|
262
|
+
# Tensor
|
|
263
|
+
def Var13(self):
|
|
264
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(30))
|
|
265
|
+
if o != 0:
|
|
266
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
267
|
+
return 0
|
|
268
|
+
|
|
269
|
+
# Tensor
|
|
270
|
+
def Var14(self):
|
|
271
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(32))
|
|
272
|
+
if o != 0:
|
|
273
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
274
|
+
return 0
|
|
275
|
+
|
|
276
|
+
# Tensor
|
|
277
|
+
def Var15(self):
|
|
278
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(34))
|
|
279
|
+
if o != 0:
|
|
280
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
281
|
+
return 0
|
|
282
|
+
|
|
283
|
+
# Tensor
|
|
284
|
+
def Var16(self):
|
|
285
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(36))
|
|
286
|
+
if o != 0:
|
|
287
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
288
|
+
return 0
|
|
289
|
+
|
|
290
|
+
# Tensor
|
|
291
|
+
def Var17(self):
|
|
292
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(38))
|
|
293
|
+
if o != 0:
|
|
294
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
295
|
+
return 0
|
|
296
|
+
|
|
297
|
+
# Tensor
|
|
298
|
+
def Index(self):
|
|
299
|
+
o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(40))
|
|
300
|
+
if o != 0:
|
|
301
|
+
return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
|
|
302
|
+
return 0
|
|
303
|
+
|
|
304
|
+
def TensorStart(builder):
|
|
305
|
+
builder.StartObject(19)
|
|
306
|
+
|
|
307
|
+
def Start(builder):
|
|
308
|
+
TensorStart(builder)
|
|
309
|
+
|
|
310
|
+
def TensorAddDataType(builder, dataType):
|
|
311
|
+
builder.PrependInt8Slot(0, dataType, 0)
|
|
312
|
+
|
|
313
|
+
def AddDataType(builder, dataType):
|
|
314
|
+
TensorAddDataType(builder, dataType)
|
|
315
|
+
|
|
316
|
+
def TensorAddVar02(builder, var02):
|
|
317
|
+
builder.PrependInt8Slot(1, var02, 0)
|
|
318
|
+
|
|
319
|
+
def AddVar02(builder, var02):
|
|
320
|
+
TensorAddVar02(builder, var02)
|
|
321
|
+
|
|
322
|
+
def TensorAddKind(builder, kind):
|
|
323
|
+
builder.PrependInt8Slot(2, kind, 0)
|
|
324
|
+
|
|
325
|
+
def AddKind(builder, kind):
|
|
326
|
+
TensorAddKind(builder, kind)
|
|
327
|
+
|
|
328
|
+
def TensorAddVar04(builder, var04):
|
|
329
|
+
builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(var04), 0)
|
|
330
|
+
|
|
331
|
+
def AddVar04(builder, var04):
|
|
332
|
+
TensorAddVar04(builder, var04)
|
|
333
|
+
|
|
334
|
+
def TensorStartVar04Vector(builder, numElems):
|
|
335
|
+
return builder.StartVector(4, numElems, 4)
|
|
336
|
+
|
|
337
|
+
def StartVar04Vector(builder, numElems):
|
|
338
|
+
return TensorStartVar04Vector(builder, numElems)
|
|
339
|
+
|
|
340
|
+
def TensorAddShape(builder, shape):
|
|
341
|
+
builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(shape), 0)
|
|
342
|
+
|
|
343
|
+
def AddShape(builder, shape):
|
|
344
|
+
TensorAddShape(builder, shape)
|
|
345
|
+
|
|
346
|
+
def TensorStartShapeVector(builder, numElems):
|
|
347
|
+
return builder.StartVector(4, numElems, 4)
|
|
348
|
+
|
|
349
|
+
def StartShapeVector(builder, numElems):
|
|
350
|
+
return TensorStartShapeVector(builder, numElems)
|
|
351
|
+
|
|
352
|
+
def TensorAddName(builder, name):
|
|
353
|
+
builder.PrependUOffsetTRelativeSlot(5, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
|
|
354
|
+
|
|
355
|
+
def AddName(builder, name):
|
|
356
|
+
TensorAddName(builder, name)
|
|
357
|
+
|
|
358
|
+
def TensorAddVar06(builder, var06):
|
|
359
|
+
builder.PrependUOffsetTRelativeSlot(6, flatbuffers.number_types.UOffsetTFlags.py_type(var06), 0)
|
|
360
|
+
|
|
361
|
+
def AddVar06(builder, var06):
|
|
362
|
+
TensorAddVar06(builder, var06)
|
|
363
|
+
|
|
364
|
+
def TensorStartVar06Vector(builder, numElems):
|
|
365
|
+
return builder.StartVector(1, numElems, 1)
|
|
366
|
+
|
|
367
|
+
def StartVar06Vector(builder, numElems):
|
|
368
|
+
return TensorStartVar06Vector(builder, numElems)
|
|
369
|
+
|
|
370
|
+
def TensorAddVar07(builder, var07):
|
|
371
|
+
builder.PrependUOffsetTRelativeSlot(7, flatbuffers.number_types.UOffsetTFlags.py_type(var07), 0)
|
|
372
|
+
|
|
373
|
+
def AddVar07(builder, var07):
|
|
374
|
+
TensorAddVar07(builder, var07)
|
|
375
|
+
|
|
376
|
+
def TensorAddVar08(builder, var08):
|
|
377
|
+
builder.PrependUOffsetTRelativeSlot(8, flatbuffers.number_types.UOffsetTFlags.py_type(var08), 0)
|
|
378
|
+
|
|
379
|
+
def AddVar08(builder, var08):
|
|
380
|
+
TensorAddVar08(builder, var08)
|
|
381
|
+
|
|
382
|
+
def TensorStartVar08Vector(builder, numElems):
|
|
383
|
+
return builder.StartVector(1, numElems, 1)
|
|
384
|
+
|
|
385
|
+
def StartVar08Vector(builder, numElems):
|
|
386
|
+
return TensorStartVar08Vector(builder, numElems)
|
|
387
|
+
|
|
388
|
+
def TensorAddVar09(builder, var09):
|
|
389
|
+
builder.PrependUOffsetTRelativeSlot(9, flatbuffers.number_types.UOffsetTFlags.py_type(var09), 0)
|
|
390
|
+
|
|
391
|
+
def AddVar09(builder, var09):
|
|
392
|
+
TensorAddVar09(builder, var09)
|
|
393
|
+
|
|
394
|
+
def TensorStartVar09Vector(builder, numElems):
|
|
395
|
+
return builder.StartVector(1, numElems, 1)
|
|
396
|
+
|
|
397
|
+
def StartVar09Vector(builder, numElems):
|
|
398
|
+
return TensorStartVar09Vector(builder, numElems)
|
|
399
|
+
|
|
400
|
+
def TensorAddVar10(builder, var10):
|
|
401
|
+
builder.PrependUOffsetTRelativeSlot(10, flatbuffers.number_types.UOffsetTFlags.py_type(var10), 0)
|
|
402
|
+
|
|
403
|
+
def AddVar10(builder, var10):
|
|
404
|
+
TensorAddVar10(builder, var10)
|
|
405
|
+
|
|
406
|
+
def TensorStartVar10Vector(builder, numElems):
|
|
407
|
+
return builder.StartVector(1, numElems, 1)
|
|
408
|
+
|
|
409
|
+
def StartVar10Vector(builder, numElems):
|
|
410
|
+
return TensorStartVar10Vector(builder, numElems)
|
|
411
|
+
|
|
412
|
+
def TensorAddVar11(builder, var11):
|
|
413
|
+
builder.PrependUOffsetTRelativeSlot(11, flatbuffers.number_types.UOffsetTFlags.py_type(var11), 0)
|
|
414
|
+
|
|
415
|
+
def AddVar11(builder, var11):
|
|
416
|
+
TensorAddVar11(builder, var11)
|
|
417
|
+
|
|
418
|
+
def TensorStartVar11Vector(builder, numElems):
|
|
419
|
+
return builder.StartVector(1, numElems, 1)
|
|
420
|
+
|
|
421
|
+
def StartVar11Vector(builder, numElems):
|
|
422
|
+
return TensorStartVar11Vector(builder, numElems)
|
|
423
|
+
|
|
424
|
+
def TensorAddSize(builder, size):
|
|
425
|
+
builder.PrependInt32Slot(12, size, 0)
|
|
426
|
+
|
|
427
|
+
def AddSize(builder, size):
|
|
428
|
+
TensorAddSize(builder, size)
|
|
429
|
+
|
|
430
|
+
def TensorAddVar13(builder, var13):
|
|
431
|
+
builder.PrependInt32Slot(13, var13, 0)
|
|
432
|
+
|
|
433
|
+
def AddVar13(builder, var13):
|
|
434
|
+
TensorAddVar13(builder, var13)
|
|
435
|
+
|
|
436
|
+
def TensorAddVar14(builder, var14):
|
|
437
|
+
builder.PrependInt32Slot(14, var14, 0)
|
|
438
|
+
|
|
439
|
+
def AddVar14(builder, var14):
|
|
440
|
+
TensorAddVar14(builder, var14)
|
|
441
|
+
|
|
442
|
+
def TensorAddVar15(builder, var15):
|
|
443
|
+
builder.PrependInt32Slot(15, var15, 0)
|
|
444
|
+
|
|
445
|
+
def AddVar15(builder, var15):
|
|
446
|
+
TensorAddVar15(builder, var15)
|
|
447
|
+
|
|
448
|
+
def TensorAddVar16(builder, var16):
|
|
449
|
+
builder.PrependInt32Slot(16, var16, 0)
|
|
450
|
+
|
|
451
|
+
def AddVar16(builder, var16):
|
|
452
|
+
TensorAddVar16(builder, var16)
|
|
453
|
+
|
|
454
|
+
def TensorAddVar17(builder, var17):
|
|
455
|
+
builder.PrependInt32Slot(17, var17, 0)
|
|
456
|
+
|
|
457
|
+
def AddVar17(builder, var17):
|
|
458
|
+
TensorAddVar17(builder, var17)
|
|
459
|
+
|
|
460
|
+
def TensorAddIndex(builder, index):
|
|
461
|
+
builder.PrependInt32Slot(18, index, 0)
|
|
462
|
+
|
|
463
|
+
def AddIndex(builder, index):
|
|
464
|
+
TensorAddIndex(builder, index)
|
|
465
|
+
|
|
466
|
+
def TensorEnd(builder):
|
|
467
|
+
return builder.EndObject()
|
|
468
|
+
|
|
469
|
+
def End(builder):
|
|
470
|
+
return TensorEnd(builder)
|
|
471
|
+
|
|
472
|
+
try:
|
|
473
|
+
from typing import List
|
|
474
|
+
except:
|
|
475
|
+
pass
|
|
476
|
+
|
|
477
|
+
class TensorT(object):
|
|
478
|
+
|
|
479
|
+
# TensorT
|
|
480
|
+
def __init__(
|
|
481
|
+
self,
|
|
482
|
+
dataType = 0,
|
|
483
|
+
var02 = 0,
|
|
484
|
+
kind = 0,
|
|
485
|
+
var04 = None,
|
|
486
|
+
shape = None,
|
|
487
|
+
name = None,
|
|
488
|
+
var06 = None,
|
|
489
|
+
var07 = None,
|
|
490
|
+
var08 = None,
|
|
491
|
+
var09 = None,
|
|
492
|
+
var10 = None,
|
|
493
|
+
var11 = None,
|
|
494
|
+
size = 0,
|
|
495
|
+
var13 = 0,
|
|
496
|
+
var14 = 0,
|
|
497
|
+
var15 = 0,
|
|
498
|
+
var16 = 0,
|
|
499
|
+
var17 = 0,
|
|
500
|
+
index = 0,
|
|
501
|
+
):
|
|
502
|
+
self.dataType = dataType # type: int
|
|
503
|
+
self.var02 = var02 # type: int
|
|
504
|
+
self.kind = kind # type: int
|
|
505
|
+
self.var04 = var04 # type: Optional[List[int]]
|
|
506
|
+
self.shape = shape # type: Optional[List[int]]
|
|
507
|
+
self.name = name # type: Optional[str]
|
|
508
|
+
self.var06 = var06 # type: Optional[List[int]]
|
|
509
|
+
self.var07 = var07 # type: Optional[str]
|
|
510
|
+
self.var08 = var08 # type: Optional[List[int]]
|
|
511
|
+
self.var09 = var09 # type: Optional[List[int]]
|
|
512
|
+
self.var10 = var10 # type: Optional[List[int]]
|
|
513
|
+
self.var11 = var11 # type: Optional[List[int]]
|
|
514
|
+
self.size = size # type: int
|
|
515
|
+
self.var13 = var13 # type: int
|
|
516
|
+
self.var14 = var14 # type: int
|
|
517
|
+
self.var15 = var15 # type: int
|
|
518
|
+
self.var16 = var16 # type: int
|
|
519
|
+
self.var17 = var17 # type: int
|
|
520
|
+
self.index = index # type: int
|
|
521
|
+
|
|
522
|
+
@classmethod
|
|
523
|
+
def InitFromBuf(cls, buf, pos):
|
|
524
|
+
tensor = Tensor()
|
|
525
|
+
tensor.Init(buf, pos)
|
|
526
|
+
return cls.InitFromObj(tensor)
|
|
527
|
+
|
|
528
|
+
@classmethod
|
|
529
|
+
def InitFromPackedBuf(cls, buf, pos=0):
|
|
530
|
+
n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
|
|
531
|
+
return cls.InitFromBuf(buf, pos+n)
|
|
532
|
+
|
|
533
|
+
@classmethod
|
|
534
|
+
def InitFromObj(cls, tensor):
|
|
535
|
+
x = TensorT()
|
|
536
|
+
x._UnPack(tensor)
|
|
537
|
+
return x
|
|
538
|
+
|
|
539
|
+
# TensorT
|
|
540
|
+
def _UnPack(self, tensor):
|
|
541
|
+
if tensor is None:
|
|
542
|
+
return
|
|
543
|
+
self.dataType = tensor.DataType()
|
|
544
|
+
self.var02 = tensor.Var02()
|
|
545
|
+
self.kind = tensor.Kind()
|
|
546
|
+
if not tensor.Var04IsNone():
|
|
547
|
+
if np is None:
|
|
548
|
+
self.var04 = []
|
|
549
|
+
for i in range(tensor.Var04Length()):
|
|
550
|
+
self.var04.append(tensor.Var04(i))
|
|
551
|
+
else:
|
|
552
|
+
self.var04 = tensor.Var04AsNumpy()
|
|
553
|
+
if not tensor.ShapeIsNone():
|
|
554
|
+
if np is None:
|
|
555
|
+
self.shape = []
|
|
556
|
+
for i in range(tensor.ShapeLength()):
|
|
557
|
+
self.shape.append(tensor.Shape(i))
|
|
558
|
+
else:
|
|
559
|
+
self.shape = tensor.ShapeAsNumpy()
|
|
560
|
+
self.name = tensor.Name()
|
|
561
|
+
if not tensor.Var06IsNone():
|
|
562
|
+
if np is None:
|
|
563
|
+
self.var06 = []
|
|
564
|
+
for i in range(tensor.Var06Length()):
|
|
565
|
+
self.var06.append(tensor.Var06(i))
|
|
566
|
+
else:
|
|
567
|
+
self.var06 = tensor.Var06AsNumpy()
|
|
568
|
+
self.var07 = tensor.Var07()
|
|
569
|
+
if not tensor.Var08IsNone():
|
|
570
|
+
if np is None:
|
|
571
|
+
self.var08 = []
|
|
572
|
+
for i in range(tensor.Var08Length()):
|
|
573
|
+
self.var08.append(tensor.Var08(i))
|
|
574
|
+
else:
|
|
575
|
+
self.var08 = tensor.Var08AsNumpy()
|
|
576
|
+
if not tensor.Var09IsNone():
|
|
577
|
+
if np is None:
|
|
578
|
+
self.var09 = []
|
|
579
|
+
for i in range(tensor.Var09Length()):
|
|
580
|
+
self.var09.append(tensor.Var09(i))
|
|
581
|
+
else:
|
|
582
|
+
self.var09 = tensor.Var09AsNumpy()
|
|
583
|
+
if not tensor.Var10IsNone():
|
|
584
|
+
if np is None:
|
|
585
|
+
self.var10 = []
|
|
586
|
+
for i in range(tensor.Var10Length()):
|
|
587
|
+
self.var10.append(tensor.Var10(i))
|
|
588
|
+
else:
|
|
589
|
+
self.var10 = tensor.Var10AsNumpy()
|
|
590
|
+
if not tensor.Var11IsNone():
|
|
591
|
+
if np is None:
|
|
592
|
+
self.var11 = []
|
|
593
|
+
for i in range(tensor.Var11Length()):
|
|
594
|
+
self.var11.append(tensor.Var11(i))
|
|
595
|
+
else:
|
|
596
|
+
self.var11 = tensor.Var11AsNumpy()
|
|
597
|
+
self.size = tensor.Size()
|
|
598
|
+
self.var13 = tensor.Var13()
|
|
599
|
+
self.var14 = tensor.Var14()
|
|
600
|
+
self.var15 = tensor.Var15()
|
|
601
|
+
self.var16 = tensor.Var16()
|
|
602
|
+
self.var17 = tensor.Var17()
|
|
603
|
+
self.index = tensor.Index()
|
|
604
|
+
|
|
605
|
+
# TensorT
|
|
606
|
+
def Pack(self, builder):
|
|
607
|
+
if self.var04 is not None:
|
|
608
|
+
if np is not None and type(self.var04) is np.ndarray:
|
|
609
|
+
var04 = builder.CreateNumpyVector(self.var04)
|
|
610
|
+
else:
|
|
611
|
+
TensorStartVar04Vector(builder, len(self.var04))
|
|
612
|
+
for i in reversed(range(len(self.var04))):
|
|
613
|
+
builder.PrependInt32(self.var04[i])
|
|
614
|
+
var04 = builder.EndVector()
|
|
615
|
+
if self.shape is not None:
|
|
616
|
+
if np is not None and type(self.shape) is np.ndarray:
|
|
617
|
+
shape = builder.CreateNumpyVector(self.shape)
|
|
618
|
+
else:
|
|
619
|
+
TensorStartShapeVector(builder, len(self.shape))
|
|
620
|
+
for i in reversed(range(len(self.shape))):
|
|
621
|
+
builder.PrependInt32(self.shape[i])
|
|
622
|
+
shape = builder.EndVector()
|
|
623
|
+
if self.name is not None:
|
|
624
|
+
name = builder.CreateString(self.name)
|
|
625
|
+
if self.var06 is not None:
|
|
626
|
+
if np is not None and type(self.var06) is np.ndarray:
|
|
627
|
+
var06 = builder.CreateNumpyVector(self.var06)
|
|
628
|
+
else:
|
|
629
|
+
TensorStartVar06Vector(builder, len(self.var06))
|
|
630
|
+
for i in reversed(range(len(self.var06))):
|
|
631
|
+
builder.PrependByte(self.var06[i])
|
|
632
|
+
var06 = builder.EndVector()
|
|
633
|
+
if self.var07 is not None:
|
|
634
|
+
var07 = builder.CreateString(self.var07)
|
|
635
|
+
if self.var08 is not None:
|
|
636
|
+
if np is not None and type(self.var08) is np.ndarray:
|
|
637
|
+
var08 = builder.CreateNumpyVector(self.var08)
|
|
638
|
+
else:
|
|
639
|
+
TensorStartVar08Vector(builder, len(self.var08))
|
|
640
|
+
for i in reversed(range(len(self.var08))):
|
|
641
|
+
builder.PrependByte(self.var08[i])
|
|
642
|
+
var08 = builder.EndVector()
|
|
643
|
+
if self.var09 is not None:
|
|
644
|
+
if np is not None and type(self.var09) is np.ndarray:
|
|
645
|
+
var09 = builder.CreateNumpyVector(self.var09)
|
|
646
|
+
else:
|
|
647
|
+
TensorStartVar09Vector(builder, len(self.var09))
|
|
648
|
+
for i in reversed(range(len(self.var09))):
|
|
649
|
+
builder.PrependByte(self.var09[i])
|
|
650
|
+
var09 = builder.EndVector()
|
|
651
|
+
if self.var10 is not None:
|
|
652
|
+
if np is not None and type(self.var10) is np.ndarray:
|
|
653
|
+
var10 = builder.CreateNumpyVector(self.var10)
|
|
654
|
+
else:
|
|
655
|
+
TensorStartVar10Vector(builder, len(self.var10))
|
|
656
|
+
for i in reversed(range(len(self.var10))):
|
|
657
|
+
builder.PrependByte(self.var10[i])
|
|
658
|
+
var10 = builder.EndVector()
|
|
659
|
+
if self.var11 is not None:
|
|
660
|
+
if np is not None and type(self.var11) is np.ndarray:
|
|
661
|
+
var11 = builder.CreateNumpyVector(self.var11)
|
|
662
|
+
else:
|
|
663
|
+
TensorStartVar11Vector(builder, len(self.var11))
|
|
664
|
+
for i in reversed(range(len(self.var11))):
|
|
665
|
+
builder.PrependByte(self.var11[i])
|
|
666
|
+
var11 = builder.EndVector()
|
|
667
|
+
TensorStart(builder)
|
|
668
|
+
TensorAddDataType(builder, self.dataType)
|
|
669
|
+
TensorAddVar02(builder, self.var02)
|
|
670
|
+
TensorAddKind(builder, self.kind)
|
|
671
|
+
if self.var04 is not None:
|
|
672
|
+
TensorAddVar04(builder, var04)
|
|
673
|
+
if self.shape is not None:
|
|
674
|
+
TensorAddShape(builder, shape)
|
|
675
|
+
if self.name is not None:
|
|
676
|
+
TensorAddName(builder, name)
|
|
677
|
+
if self.var06 is not None:
|
|
678
|
+
TensorAddVar06(builder, var06)
|
|
679
|
+
if self.var07 is not None:
|
|
680
|
+
TensorAddVar07(builder, var07)
|
|
681
|
+
if self.var08 is not None:
|
|
682
|
+
TensorAddVar08(builder, var08)
|
|
683
|
+
if self.var09 is not None:
|
|
684
|
+
TensorAddVar09(builder, var09)
|
|
685
|
+
if self.var10 is not None:
|
|
686
|
+
TensorAddVar10(builder, var10)
|
|
687
|
+
if self.var11 is not None:
|
|
688
|
+
TensorAddVar11(builder, var11)
|
|
689
|
+
TensorAddSize(builder, self.size)
|
|
690
|
+
TensorAddVar13(builder, self.var13)
|
|
691
|
+
TensorAddVar14(builder, self.var14)
|
|
692
|
+
TensorAddVar15(builder, self.var15)
|
|
693
|
+
TensorAddVar16(builder, self.var16)
|
|
694
|
+
TensorAddVar17(builder, self.var17)
|
|
695
|
+
TensorAddIndex(builder, self.index)
|
|
696
|
+
tensor = TensorEnd(builder)
|
|
697
|
+
return tensor
|