Skip to content

Commit 83bfbf4

Browse files
committed
evquantize: saturate method typo fix.
1 parent b900259 commit 83bfbf4

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

src/caffe/layers/conv_layer.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ void ConvolutionLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
3535
caffe_gpu_scal(count_t, output_scale, top_data);
3636
caffe_gpu_round(count_t, top_data);
3737
}
38-
if(saturate == ReLUParameter_SaturateMethod_Signed)
38+
if(saturate == ConvolutionParameter_SaturateMethod_Signed)
3939
caffe_gpu_signed_saturate(count_t, top_data);
40-
if(saturate == ReLUParameter_SaturateMethod_Unsigned)
40+
if(saturate == ConvolutionParameter_SaturateMethod_Unsigned)
4141
caffe_gpu_unsigned_saturate(count_t, top_data);
4242
//CUSTOMIZATION-->
4343
}

src/caffe/layers/deconv_layer.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,9 @@ void DeconvolutionLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
3434
caffe_gpu_scal(count_t, output_scale, top_data);
3535
caffe_gpu_round(count_t, top_data);
3636
}
37-
if(saturate == ReLUParameter_SaturateMethod_Signed)
37+
if(saturate == ConvolutionParameter_SaturateMethod_Signed)
3838
caffe_gpu_signed_saturate(count_t, top_data);
39-
if(saturate == ReLUParameter_SaturateMethod_Unsigned)
39+
if(saturate == ConvolutionParameter_SaturateMethod_Unsigned)
4040
caffe_gpu_unsigned_saturate(count_t, top_data);
4141
//CUSTOMIZATION-->
4242
}

src/caffe/layers/eltwise_layer.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,9 @@ void EltwiseLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
5555
caffe_gpu_scal(count, output_scale_, top_data);
5656
caffe_gpu_round(count, top_data);
5757
}
58-
if(saturate_ == ReLUParameter_SaturateMethod_Signed)
58+
if(saturate_ == EltwiseParameter_SaturateMethod_Signed)
5959
caffe_gpu_signed_saturate(count, top_data);
60-
if(saturate_ == ReLUParameter_SaturateMethod_Unsigned)
60+
if(saturate_ == EltwiseParameter_SaturateMethod_Unsigned)
6161
caffe_gpu_unsigned_saturate(count, top_data);
6262
//CUSTOMIZATION-->
6363
break;

0 commit comments

Comments
 (0)