-
Notifications
You must be signed in to change notification settings - Fork 110
/
FullyConnectedOptions.go
101 lines (85 loc) · 3.08 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
// Code generated by the FlatBuffers compiler. DO NOT EDIT.
package tflite
import (
flatbuffers "github.com/google/flatbuffers/go"
)
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 GetSizePrefixedRootAsFullyConnectedOptions(buf []byte, offset flatbuffers.UOffsetT) *FullyConnectedOptions {
n := flatbuffers.GetUOffsetT(buf[offset+flatbuffers.SizeUint32:])
x := &FullyConnectedOptions{}
x.Init(buf, n+offset+flatbuffers.SizeUint32)
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()
}