Upload quant_config.json
Browse files- quant_config.json +74 -48
quant_config.json
CHANGED
@@ -2,88 +2,114 @@
|
|
2 |
"per_channel": false,
|
3 |
"reduce_range": false,
|
4 |
"per_model_config": {
|
5 |
-
"
|
6 |
"op_types": [
|
7 |
-
"Reshape",
|
8 |
-
"Shape",
|
9 |
-
"Unsqueeze",
|
10 |
"Div",
|
11 |
-
"Sub",
|
12 |
-
"Erf",
|
13 |
-
"Gather",
|
14 |
-
"Sqrt",
|
15 |
"Concat",
|
16 |
-
"
|
|
|
|
|
17 |
"MatMul",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
"Constant",
|
19 |
"Pow",
|
20 |
-
"ReduceMean",
|
21 |
-
"Transpose",
|
22 |
-
"Mul",
|
23 |
"Softmax",
|
24 |
-
"
|
|
|
|
|
|
|
|
|
25 |
],
|
26 |
-
"weight_type": "
|
27 |
},
|
28 |
-
"
|
29 |
"op_types": [
|
30 |
-
"
|
31 |
"Gather",
|
32 |
-
"Cast",
|
33 |
-
"ConstantOfShape",
|
34 |
-
"Transpose",
|
35 |
-
"Reshape",
|
36 |
-
"Sqrt",
|
37 |
-
"Softmax",
|
38 |
-
"Mul",
|
39 |
"Concat",
|
40 |
-
"Unsqueeze",
|
41 |
-
"Where",
|
42 |
"Less",
|
|
|
43 |
"MatMul",
|
44 |
-
"
|
45 |
-
"Shape",
|
46 |
-
"Range",
|
47 |
-
"Div",
|
48 |
-
"Squeeze",
|
49 |
"Add",
|
50 |
"Expand",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
"Constant",
|
52 |
"Pow",
|
53 |
-
"
|
54 |
-
"
|
55 |
-
"
|
|
|
|
|
|
|
56 |
],
|
57 |
"weight_type": "QInt8"
|
58 |
},
|
59 |
-
"
|
60 |
"op_types": [
|
61 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
],
|
63 |
"weight_type": "QInt8"
|
64 |
},
|
65 |
-
"
|
66 |
"op_types": [
|
67 |
-
"Reshape",
|
68 |
-
"Shape",
|
69 |
-
"Unsqueeze",
|
70 |
"Div",
|
71 |
-
"Erf",
|
72 |
"Gather",
|
73 |
-
"
|
74 |
-
"Add",
|
75 |
-
"MatMul",
|
76 |
"Constant",
|
77 |
-
"Slice",
|
78 |
"Pow",
|
79 |
-
"ReduceMean",
|
80 |
"Sub",
|
|
|
81 |
"Mul",
|
82 |
-
"
|
83 |
"Transpose",
|
84 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
85 |
],
|
86 |
-
"weight_type": "
|
87 |
}
|
88 |
}
|
89 |
}
|
|
|
2 |
"per_channel": false,
|
3 |
"reduce_range": false,
|
4 |
"per_model_config": {
|
5 |
+
"decoder_model": {
|
6 |
"op_types": [
|
|
|
|
|
|
|
7 |
"Div",
|
|
|
|
|
|
|
|
|
8 |
"Concat",
|
9 |
+
"Gather",
|
10 |
+
"Less",
|
11 |
+
"ReduceMean",
|
12 |
"MatMul",
|
13 |
+
"Erf",
|
14 |
+
"Add",
|
15 |
+
"Expand",
|
16 |
+
"Sub",
|
17 |
+
"ConstantOfShape",
|
18 |
+
"Range",
|
19 |
+
"Sqrt",
|
20 |
+
"Slice",
|
21 |
+
"Squeeze",
|
22 |
+
"Cast",
|
23 |
+
"Equal",
|
24 |
+
"Where",
|
25 |
"Constant",
|
26 |
"Pow",
|
|
|
|
|
|
|
27 |
"Softmax",
|
28 |
+
"Shape",
|
29 |
+
"Reshape",
|
30 |
+
"Mul",
|
31 |
+
"Transpose",
|
32 |
+
"Unsqueeze"
|
33 |
],
|
34 |
+
"weight_type": "QInt8"
|
35 |
},
|
36 |
+
"decoder_model_merged": {
|
37 |
"op_types": [
|
38 |
+
"Div",
|
39 |
"Gather",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
"Concat",
|
|
|
|
|
41 |
"Less",
|
42 |
+
"ReduceMean",
|
43 |
"MatMul",
|
44 |
+
"Erf",
|
|
|
|
|
|
|
|
|
45 |
"Add",
|
46 |
"Expand",
|
47 |
+
"Sub",
|
48 |
+
"ConstantOfShape",
|
49 |
+
"Range",
|
50 |
+
"Sqrt",
|
51 |
+
"Slice",
|
52 |
+
"Squeeze",
|
53 |
+
"Cast",
|
54 |
+
"Equal",
|
55 |
+
"Where",
|
56 |
+
"If",
|
57 |
"Constant",
|
58 |
"Pow",
|
59 |
+
"Softmax",
|
60 |
+
"Shape",
|
61 |
+
"Reshape",
|
62 |
+
"Mul",
|
63 |
+
"Transpose",
|
64 |
+
"Unsqueeze"
|
65 |
],
|
66 |
"weight_type": "QInt8"
|
67 |
},
|
68 |
+
"decoder_with_past_model": {
|
69 |
"op_types": [
|
70 |
+
"Div",
|
71 |
+
"Concat",
|
72 |
+
"Gather",
|
73 |
+
"Constant",
|
74 |
+
"Pow",
|
75 |
+
"Sub",
|
76 |
+
"Mul",
|
77 |
+
"Transpose",
|
78 |
+
"Add",
|
79 |
+
"MatMul",
|
80 |
+
"Softmax",
|
81 |
+
"Sqrt",
|
82 |
+
"Slice",
|
83 |
+
"Shape",
|
84 |
+
"Reshape",
|
85 |
+
"ReduceMean",
|
86 |
+
"Erf",
|
87 |
+
"Unsqueeze"
|
88 |
],
|
89 |
"weight_type": "QInt8"
|
90 |
},
|
91 |
+
"encoder_model": {
|
92 |
"op_types": [
|
|
|
|
|
|
|
93 |
"Div",
|
|
|
94 |
"Gather",
|
95 |
+
"Concat",
|
|
|
|
|
96 |
"Constant",
|
|
|
97 |
"Pow",
|
|
|
98 |
"Sub",
|
99 |
+
"Unsqueeze",
|
100 |
"Mul",
|
101 |
+
"Conv",
|
102 |
"Transpose",
|
103 |
+
"MatMul",
|
104 |
+
"Softmax",
|
105 |
+
"Sqrt",
|
106 |
+
"Shape",
|
107 |
+
"Reshape",
|
108 |
+
"ReduceMean",
|
109 |
+
"Erf",
|
110 |
+
"Add"
|
111 |
],
|
112 |
+
"weight_type": "QUInt8"
|
113 |
}
|
114 |
}
|
115 |
}
|