From a5b8693369e7e6be16f4cd769b62a6d6330fdefa Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Thu, 26 May 2016 09:06:34 -0800 Subject: [PATCH] Made the pooling code fp16 friendly Change: 123329817 --- tensorflow/core/kernels/pooling_ops_common.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/core/kernels/pooling_ops_common.h b/tensorflow/core/kernels/pooling_ops_common.h index 4bdeb6bf9c5..f38e8e751c5 100644 --- a/tensorflow/core/kernels/pooling_ops_common.h +++ b/tensorflow/core/kernels/pooling_ops_common.h @@ -305,7 +305,7 @@ void SpatialAvgPool(OpKernelContext* context, Tensor* output, const int out_offset = (b * params.out_height + ph) * params.out_width + pw; out_mat.col(out_offset) += in_mat.col(in_offset); - out_count(out_offset)++; + out_count(out_offset) += T(1); } } }