add missing saturate in relu quantize compute

foss-for-synopsys-dwc-arc-processors · Aug 3, 2021 · 3c5f610 · 3c5f610
1 parent 3c2d549
commit 3c5f610
Showing 1 changed file with 1 addition and 0 deletions.
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(),