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 --- arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'arm_compute/dynamic_fusion') diff --git a/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h b/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h index 16d88af570..6e1bcdbbfd 100644 --- a/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h +++ b/arm_compute/dynamic_fusion/sketch/gpu/operators/GpuPool2d.h @@ -48,8 +48,15 @@ public: /* Set mixed_precision */ GpuPool2dSettings &mixed_precision(bool mixed_precision); + /* Get using -infinity as limit flag */ + bool use_inf_as_limit() const; + + /* Set using -infinity as limit flag */ + GpuPool2dSettings use_inf_as_limit(bool use_inf_as_limit); + private: bool _mixed_precision{ false }; + bool _use_inf_as_limit{ true }; }; /** Operator interface. */ -- cgit v1.2.1