Skip to content

Commit ef00b5e

Browse files
committed
add scale/zero point for relu6 quantize handling
1 parent 6af56be commit ef00b5e

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

FEATURES.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ InnerProductParameter
131131
ConcatParameter
132132
EltwiseParameter
133133
PoolingParameter
134+
ReLUParameter
134135
SigmoidParameter
135136
SoftmaxParameter
136137

src/caffe/proto/caffe.proto

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2326,6 +2326,10 @@ message ReLUParameter {
23262326
optional bool relu6 = 3 [default = false]; //CUSTOMIZATION, if set true, the output is min(max(0, input), 6)
23272327
optional float maximum = 6 [default = 0]; //CUSTOMIZATION, act as MAX range setting
23282328
optional float minimum = 7 [default = 0]; //CUSTOMIZATION, act as MIN range setting
2329+
optional float input_scale = 13 [default = 1]; //CUSTOMIZATION
2330+
optional float output_scale = 14 [default = 1]; //CUSTOMIZATION
2331+
optional int32 input_zero_point = 15 [default = 0]; //CUSTOMIZATION
2332+
optional int32 output_zero_point = 16 [default = 0]; //CUSTOMIZATION
23292333
//<--CUSTOMIZATION
23302334
enum SaturateMethod {
23312335
None = 0;

0 commit comments

Comments
 (0)