add in/output zero-point param for softmax

vera121 · vera121 · commit 159061550d15 · 2021-03-31T12:30:16.000+02:00
diff --git a/src/caffe/proto/caffe.proto b/src/caffe/proto/caffe.proto
@@ -2540,6 +2540,8 @@ message SoftmaxParameter {
   optional int32 axis = 2 [default = 1];
   optional double input_scale = 3 [default = 1]; //CUSTOMIZATION
   optional double output_scale = 4 [default = 1]; //CUSTOMIZATION
+  optional int32 input_zero_point = 5 [default = 0]; //CUSTOMIZATION
+  optional int32 output_zero_point = 6 [default = 0]; //CUSTOMIZATION
 }
 
 message LogSoftmaxParameter {

Original file line number	Diff line number	Diff line change
`@@ -2540,6 +2540,8 @@ message SoftmaxParameter {`
`2540`	`2540`	`optional int32 axis = 2 [default = 1];`
`2541`	`2541`	`optional double input_scale = 3 [default = 1]; //CUSTOMIZATION`
`2542`	`2542`	`optional double output_scale = 4 [default = 1]; //CUSTOMIZATION`
	`2543`	`+ optional int32 input_zero_point = 5 [default = 0]; //CUSTOMIZATION`
	`2544`	`+ optional int32 output_zero_point = 6 [default = 0]; //CUSTOMIZATION`
`2543`	`2545`	`}`
`2544`	`2546`
`2545`	`2547`	`message LogSoftmaxParameter {`