Made the pooling code fp16 friendly

Change: 123329817
2016-05-26 09:06:34 -08:00 · 2016-05-26 09:06:34 -08:00 · a5b8693369
commit a5b8693369
parent a2009115df
1 changed files with 1 additions and 1 deletions
--- 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);
          }
        }
      }