inference acc: add round to even for average pooling output.

vera121 · vera121 · commit 249502dc72b2 · 2018-09-06T05:03:54.000-04:00
diff --git a/src/caffe/layers/pooling_layer.cu b/src/caffe/layers/pooling_layer.cu
@@ -85,6 +85,7 @@ __global__ void AvePoolForward(const int nthreads,
     }
     if (output_shift_instead_division != Dtype(0)) {
       top_data[index] = aveval / output_shift_instead_division;
+      top_data[index] = rint(top_data[index]);
     }
     else{
       top_data[index] = aveval / pool_size;
@@ -136,6 +137,7 @@ __global__ void AvePoolForward_TF(const int nthreads,
         top_data[index] = aveval / output_shift_instead_division;
       else
         top_data[index] = aveval / output_shift_instead_division * full_pool_size / pool_size;
+      top_data[index] = rint(top_data[index]);
     }
     else{
       top_data[index] = aveval / pool_size;

Original file line number	Diff line number	Diff line change
`@@ -85,6 +85,7 @@ __global__ void AvePoolForward(const int nthreads,`
`85`	`85`	`}`
`86`	`86`	`if (output_shift_instead_division != Dtype(0)) {`
`87`	`87`	`top_data[index] = aveval / output_shift_instead_division;`
	`88`	`+ top_data[index] = rint(top_data[index]);`
`88`	`89`	`}`
`89`	`90`	`else{`
`90`	`91`	`top_data[index] = aveval / pool_size;`
`@@ -136,6 +137,7 @@ __global__ void AvePoolForward_TF(const int nthreads,`
`136`	`137`	`top_data[index] = aveval / output_shift_instead_division;`
`137`	`138`	`else`
`138`	`139`	`top_data[index] = aveval / output_shift_instead_division * full_pool_size / pool_size;`
	`140`	`+ top_data[index] = rint(top_data[index]);`
`139`	`141`	`}`
`140`	`142`	`else{`
`141`	`143`	`top_data[index] = aveval / pool_size;`