Skip to content

Commit b49d702

Browse files
author
snoopyisadog
committed
add quant_params for weight/bias in Convolution&InnerProduct
1 parent dd85dae commit b49d702

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

src/caffe/proto/caffe.proto

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1370,8 +1370,12 @@ message ConvolutionParameter {
13701370

13711371
optional double input_scale = 30 [default = 1]; //CUSTOMIZATION
13721372
optional double output_scale = 31 [default = 1]; //CUSTOMIZATION
1373-
optional int32 input_zero_point = 33 [default = 0]; //CUSTOMIZATION
1374-
optional int32 output_zero_point = 34 [default = 0]; //CUSTOMIZATION
1373+
optional double weight_scale = 33 [default = 1]; //CUSTOMIZATION
1374+
optional double bias_scale = 34 [default = 1]; //CUSTOMIZATION
1375+
optional int32 input_zero_point = 35 [default = 0]; //CUSTOMIZATION
1376+
optional int32 output_zero_point = 36 [default = 0]; //CUSTOMIZATION
1377+
optional int32 weight_zero_point = 37 [default = 0]; //CUSTOMIZATION
1378+
optional int32 bias_zero_point = 38 [default = 0]; //CUSTOMIZATION
13751379
//<--CUSTOMIZATION
13761380
enum SaturateMethod {
13771381
None = 0;
@@ -1867,7 +1871,14 @@ message InnerProductParameter {
18671871
optional bool dis_mode = 8 [default = false];
18681872
optional bool gen_mode = 9 [default = false];
18691873

1870-
optional double output_scale = 11 [default = 1]; //CUSTOMIZATION
1874+
optional double input_scale = 11 [default = 1]; //CUSTOMIZATION
1875+
optional double output_scale = 12 [default = 1]; //CUSTOMIZATION
1876+
optional double weight_scale = 13 [default = 1]; //CUSTOMIZATION
1877+
optional double bias_scale = 14 [default = 1]; //CUSTOMIZATION
1878+
optional int32 input_zero_point = 15 [default = 0]; //CUSTOMIZATION
1879+
optional int32 output_zero_point = 16 [default = 0]; //CUSTOMIZATION
1880+
optional int32 weight_zero_point = 17 [default = 0]; //CUSTOMIZATION
1881+
optional int32 bias_zero_point = 18 [default = 0]; //CUSTOMIZATION
18711882
}
18721883

18731884
message InputParameter {

0 commit comments

Comments
 (0)