From 40a245f943330219e59adaf9c0d23139d674a5b5 Mon Sep 17 00:00:00 2001 From: SiCongLi Date: Thu, 2 Dec 2021 11:54:32 +0000 Subject: Unify Cpu Max pool 2d minimum value to be -inf for floating point Resolves: COMPMID-4998 Signed-off-by: SiCongLi Change-Id: Ia34c4b063ebcdb193db0a3d6c22f6769a8a902d5 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6822 Reviewed-by: Gian Marco Iodice Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/cpu/kernels/pool2d/neon/fp32.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/cpu/kernels/pool2d/neon/fp32.cpp') diff --git a/src/cpu/kernels/pool2d/neon/fp32.cpp b/src/cpu/kernels/pool2d/neon/fp32.cpp index 26a32ed9d4..e4261f746d 100644 --- a/src/cpu/kernels/pool2d/neon/fp32.cpp +++ b/src/cpu/kernels/pool2d/neon/fp32.cpp @@ -223,7 +223,7 @@ void poolingMxN_fp32_neon_nhwc(const ITensor *src, ITensor *dst0, ITensor *dst1, } else { - vres = vdupq_n_f32(std::numeric_limits::lowest()); + vres = vdupq_n_f32(-std::numeric_limits::infinity()); for(int y = pool_start_y; y < pool_end_y; ++y) { for(int x = pool_start_x; x < pool_end_x; ++x) @@ -285,7 +285,7 @@ void poolingMxN_fp32_neon_nhwc(const ITensor *src, ITensor *dst0, ITensor *dst1, } else { - res = std::numeric_limits::lowest(); + res = -std::numeric_limits::infinity(); for(int y = pool_start_y; y < pool_end_y; ++y) { for(int x = pool_start_x; x < pool_end_x; ++x) -- cgit v1.2.1