From 227db8db83cd85d8704a8edbd4e8c88af0dd1f84 Mon Sep 17 00:00:00 2001 From: Adnan AlSinan Date: Tue, 14 Feb 2023 14:24:09 +0000 Subject: Add an option to use lowest for max-pooling - Add a parameter in PoolingLayerInfo class to pick which value to use as min for max-pooling. Resolves: [ONCPUML-1166] Signed-off-by: Adnan AlSinan Change-Id: I34e1cccc15176bbf31523c61e99f3188ddca23e1 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8989 Comments-Addressed: Arm Jenkins Reviewed-by: SiCong Li Reviewed-by: Pablo Marquez Tello Tested-by: Arm Jenkins Benchmark: Arm Jenkins --- tests/validation/reference/PoolingLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/validation/reference') diff --git a/tests/validation/reference/PoolingLayer.cpp b/tests/validation/reference/PoolingLayer.cpp index 378d91d829..6a358ced0c 100644 --- a/tests/validation/reference/PoolingLayer.cpp +++ b/tests/validation/reference/PoolingLayer.cpp @@ -87,7 +87,7 @@ SimpleTensor pooling_layer_internal(const SimpleTensor &src, const Pooling int hend = std::min(hstart + pool_size_y, h_src); wstart = std::max(wstart, 0); hstart = std::max(hstart, 0); - auto max_val = -std::numeric_limits::infinity(); + auto max_val = info.use_inf_as_limit ? -std::numeric_limits::infinity() : std::numeric_limits::lowest(); int max_index{ 0 }; for(int y = hstart; y < hend; ++y) { -- cgit v1.2.1