add missing saturate in relu quantize compute

vera121 · vera121 · commit 3c5f6106984e · 2021-08-03T11:41:25.000+02:00
diff --git a/src/caffe/layers/relu_layer.cpp b/src/caffe/layers/relu_layer.cpp
@@ -83,6 +83,7 @@ void ReLULayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
   if (quant_out) {
     // do not reuse "top_data"; it is shifted during the computation
     caffe_cpu_quantize<Dtype>(top[0]->count(), top[0]->mutable_cpu_data(), output_scale_, output_zero_point_);
+    caffe_cpu_saturate(top[0]->count(), top[0]->mutable_cpu_data(), saturate_);
   }
   if (quant_in) {
     caffe_cpu_quantize<Dtype>(bottom[0]->count(), bottom[0]->mutable_cpu_data(),

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,7 @@ void ReLULayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,`
`83`	`83`	`if (quant_out) {`
`84`	`84`	`// do not reuse "top_data"; it is shifted during the computation`
`85`	`85`	`caffe_cpu_quantize<Dtype>(top[0]->count(), top[0]->mutable_cpu_data(), output_scale_, output_zero_point_);`
	`86`	`+ caffe_cpu_saturate(top[0]->count(), top[0]->mutable_cpu_data(), saturate_);`
`86`	`87`	`}`
`87`	`88`	`if (quant_in) {`
`88`	`89`	`caffe_cpu_quantize<Dtype>(bottom[0]->count(), bottom[0]->mutable_cpu_data(),`