/
FullyConnectedOptions.go
125 lines (106 loc) · 3.9 KB
/
FullyConnectedOptions.go
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
// Code generated by the FlatBuffers compiler. DO NOT EDIT.
package tflite
import (
flatbuffers "github.com/google/flatbuffers/go"
)
type FullyConnectedOptionsT struct {
FusedActivationFunction ActivationFunctionType
WeightsFormat FullyConnectedOptionsWeightsFormat
KeepNumDims bool
AsymmetricQuantizeInputs bool
}
func (t *FullyConnectedOptionsT) Pack(builder *flatbuffers.Builder) flatbuffers.UOffsetT {
if t == nil { return 0 }
FullyConnectedOptionsStart(builder)
FullyConnectedOptionsAddFusedActivationFunction(builder, t.FusedActivationFunction)
FullyConnectedOptionsAddWeightsFormat(builder, t.WeightsFormat)
FullyConnectedOptionsAddKeepNumDims(builder, t.KeepNumDims)
FullyConnectedOptionsAddAsymmetricQuantizeInputs(builder, t.AsymmetricQuantizeInputs)
return FullyConnectedOptionsEnd(builder)
}
func (rcv *FullyConnectedOptions) UnPackTo(t *FullyConnectedOptionsT) {
t.FusedActivationFunction = rcv.FusedActivationFunction()
t.WeightsFormat = rcv.WeightsFormat()
t.KeepNumDims = rcv.KeepNumDims()
t.AsymmetricQuantizeInputs = rcv.AsymmetricQuantizeInputs()
}
func (rcv *FullyConnectedOptions) UnPack() *FullyConnectedOptionsT {
if rcv == nil { return nil }
t := &FullyConnectedOptionsT{}
rcv.UnPackTo(t)
return t
}
type FullyConnectedOptions struct {
_tab flatbuffers.Table
}
func GetRootAsFullyConnectedOptions(buf []byte, offset flatbuffers.UOffsetT) *FullyConnectedOptions {
n := flatbuffers.GetUOffsetT(buf[offset:])
x := &FullyConnectedOptions{}
x.Init(buf, n+offset)
return x
}
func (rcv *FullyConnectedOptions) Init(buf []byte, i flatbuffers.UOffsetT) {
rcv._tab.Bytes = buf
rcv._tab.Pos = i
}
func (rcv *FullyConnectedOptions) Table() flatbuffers.Table {
return rcv._tab
}
func (rcv *FullyConnectedOptions) FusedActivationFunction() ActivationFunctionType {
o := flatbuffers.UOffsetT(rcv._tab.Offset(4))
if o != 0 {
return ActivationFunctionType(rcv._tab.GetInt8(o + rcv._tab.Pos))
}
return 0
}
func (rcv *FullyConnectedOptions) MutateFusedActivationFunction(n ActivationFunctionType) bool {
return rcv._tab.MutateInt8Slot(4, int8(n))
}
func (rcv *FullyConnectedOptions) WeightsFormat() FullyConnectedOptionsWeightsFormat {
o := flatbuffers.UOffsetT(rcv._tab.Offset(6))
if o != 0 {
return FullyConnectedOptionsWeightsFormat(rcv._tab.GetInt8(o + rcv._tab.Pos))
}
return 0
}
func (rcv *FullyConnectedOptions) MutateWeightsFormat(n FullyConnectedOptionsWeightsFormat) bool {
return rcv._tab.MutateInt8Slot(6, int8(n))
}
func (rcv *FullyConnectedOptions) KeepNumDims() bool {
o := flatbuffers.UOffsetT(rcv._tab.Offset(8))
if o != 0 {
return rcv._tab.GetBool(o + rcv._tab.Pos)
}
return false
}
func (rcv *FullyConnectedOptions) MutateKeepNumDims(n bool) bool {
return rcv._tab.MutateBoolSlot(8, n)
}
func (rcv *FullyConnectedOptions) AsymmetricQuantizeInputs() bool {
o := flatbuffers.UOffsetT(rcv._tab.Offset(10))
if o != 0 {
return rcv._tab.GetBool(o + rcv._tab.Pos)
}
return false
}
func (rcv *FullyConnectedOptions) MutateAsymmetricQuantizeInputs(n bool) bool {
return rcv._tab.MutateBoolSlot(10, n)
}
func FullyConnectedOptionsStart(builder *flatbuffers.Builder) {
builder.StartObject(4)
}
func FullyConnectedOptionsAddFusedActivationFunction(builder *flatbuffers.Builder, fusedActivationFunction ActivationFunctionType) {
builder.PrependInt8Slot(0, int8(fusedActivationFunction), 0)
}
func FullyConnectedOptionsAddWeightsFormat(builder *flatbuffers.Builder, weightsFormat FullyConnectedOptionsWeightsFormat) {
builder.PrependInt8Slot(1, int8(weightsFormat), 0)
}
func FullyConnectedOptionsAddKeepNumDims(builder *flatbuffers.Builder, keepNumDims bool) {
builder.PrependBoolSlot(2, keepNumDims, false)
}
func FullyConnectedOptionsAddAsymmetricQuantizeInputs(builder *flatbuffers.Builder, asymmetricQuantizeInputs bool) {
builder.PrependBoolSlot(3, asymmetricQuantizeInputs, false)
}
func FullyConnectedOptionsEnd(builder *flatbuffers.Builder) flatbuffers.UOffsetT {
return builder.EndObject()
}