enums.py Source File

enums.py Source File#

Mobilint SDK qb Compiler: enums.py Source File
Mobilint SDK qb Compiler v1.2
MCS002-KR
enums.py
1"""Auto-generated enum definitions from config_schema.yaml."""
2
3from enum import Enum
4
5
6class Device(str, Enum):
7 gpu = "gpu"
8 cpu = "cpu"
9
10 def __str__(self) -> str:
11 return self.value
12
13
14class InferenceScheme(str, Enum):
15 single = "single"
16 multi = "multi"
17 _global = "global"
18 global4 = "global4"
19 global8 = "global8"
20 all = "all"
21
22 def __str__(self) -> str:
23 return self.value
24
25
26class LutClusteringMethod(str, Enum):
27 Scale = "Scale"
28 Scale2D = "Scale2D"
29 ReparameterizedL2 = "ReparameterizedL2"
30 SubgraphIoU = "SubgraphIoU"
31 DomainAwareL2 = "DomainAwareL2"
32 OverlapWeightedL2 = "OverlapWeightedL2"
33 Fast = "Fast"
34
35 def __str__(self) -> str:
36 return self.value
37
38
39class Method(str, Enum):
40 DeleteFloat = "DeleteFloat"
41 SaveFloat = "SaveFloat"
42 MoveFloat = "MoveFloat"
43 KeepFloat = "KeepFloat"
44 KeepAll = "KeepAll"
45
46 def __str__(self) -> str:
47 return self.value
48
49
50class MoeExpertSelectionMode(str, Enum):
51 TopK = "TopK"
52 All = "All"
53 Threshold = "Threshold"
54
55 def __str__(self) -> str:
56 return self.value
57
58
59class QuantMethod(str, Enum):
60 WChALayer = "WChALayer"
61 WChAMulti = "WChAMulti"
62 WChALayerZeropoint = "WChALayerZeropoint"
63 WChAMultiZeropoint = "WChAMultiZeropoint"
64
65 def __str__(self) -> str:
66 return self.value
67
68
69class QuantMode(str, Enum):
70 Max = "Max"
71 MaxPercentile = "MaxPercentile"
72 Histogram = "Histogram"
73
74 def __str__(self) -> str:
75 return self.value
76
77
78class QuantOutput(str, Enum):
79 Layer = "Layer"
80 Ch = "Ch"
81 Sigmoid = "Sigmoid"
82
83 def __str__(self) -> str:
84 return self.value
85
86
87class SearchType(str, Enum):
88 Percentile = "Percentile"
89 MSE = "MSE"
90 KL = "KL"
91
92 def __str__(self) -> str:
93 return self.value
Generated by doxygen 1.14.0