-
Notifications
You must be signed in to change notification settings - Fork 110
/
TensorType.go
74 lines (67 loc) · 2.11 KB
/
TensorType.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
// Code generated by the FlatBuffers compiler. DO NOT EDIT.
package tflite
import "strconv"
type TensorType int8
const (
TensorTypeFLOAT32 TensorType = 0
TensorTypeFLOAT16 TensorType = 1
TensorTypeINT32 TensorType = 2
TensorTypeUINT8 TensorType = 3
TensorTypeINT64 TensorType = 4
TensorTypeSTRING TensorType = 5
TensorTypeBOOL TensorType = 6
TensorTypeINT16 TensorType = 7
TensorTypeCOMPLEX64 TensorType = 8
TensorTypeINT8 TensorType = 9
TensorTypeFLOAT64 TensorType = 10
TensorTypeCOMPLEX128 TensorType = 11
TensorTypeUINT64 TensorType = 12
TensorTypeRESOURCE TensorType = 13
TensorTypeVARIANT TensorType = 14
TensorTypeUINT32 TensorType = 15
TensorTypeUINT16 TensorType = 16
)
var EnumNamesTensorType = map[TensorType]string{
TensorTypeFLOAT32: "FLOAT32",
TensorTypeFLOAT16: "FLOAT16",
TensorTypeINT32: "INT32",
TensorTypeUINT8: "UINT8",
TensorTypeINT64: "INT64",
TensorTypeSTRING: "STRING",
TensorTypeBOOL: "BOOL",
TensorTypeINT16: "INT16",
TensorTypeCOMPLEX64: "COMPLEX64",
TensorTypeINT8: "INT8",
TensorTypeFLOAT64: "FLOAT64",
TensorTypeCOMPLEX128: "COMPLEX128",
TensorTypeUINT64: "UINT64",
TensorTypeRESOURCE: "RESOURCE",
TensorTypeVARIANT: "VARIANT",
TensorTypeUINT32: "UINT32",
TensorTypeUINT16: "UINT16",
}
var EnumValuesTensorType = map[string]TensorType{
"FLOAT32": TensorTypeFLOAT32,
"FLOAT16": TensorTypeFLOAT16,
"INT32": TensorTypeINT32,
"UINT8": TensorTypeUINT8,
"INT64": TensorTypeINT64,
"STRING": TensorTypeSTRING,
"BOOL": TensorTypeBOOL,
"INT16": TensorTypeINT16,
"COMPLEX64": TensorTypeCOMPLEX64,
"INT8": TensorTypeINT8,
"FLOAT64": TensorTypeFLOAT64,
"COMPLEX128": TensorTypeCOMPLEX128,
"UINT64": TensorTypeUINT64,
"RESOURCE": TensorTypeRESOURCE,
"VARIANT": TensorTypeVARIANT,
"UINT32": TensorTypeUINT32,
"UINT16": TensorTypeUINT16,
}
func (v TensorType) String() string {
if s, ok := EnumNamesTensorType[v]; ok {
return s
}
return "TensorType(" + strconv.FormatInt(int64(v), 10) + ")"
}