From fe56edb4fd7a620fea4b6002d87a9763bdf8791a Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Tue, 13 Apr 2021 20:21:11 +0100 Subject: Add padding consideration to pooling index computation Fix the pooling kernel which has been missing consideration of left padding, which can be implictly added by external kernels. Additionally, tests for FP16 have been added for the logic. Resolves: COMPMID-4363 Change-Id: I5655991cb80f749fb1ae9bbd3918b436a078f5d1 Signed-off-by: Sang-Hoon Park Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5421 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Giorgio Arena Comments-Addressed: Arm Jenkins --- tests/validation/fixtures/PoolingLayerFixture.h | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'tests/validation/fixtures/PoolingLayerFixture.h') diff --git a/tests/validation/fixtures/PoolingLayerFixture.h b/tests/validation/fixtures/PoolingLayerFixture.h index 66e09d5bdb..74ba3b3431 100644 --- a/tests/validation/fixtures/PoolingLayerFixture.h +++ b/tests/validation/fixtures/PoolingLayerFixture.h @@ -114,8 +114,7 @@ protected: ARM_COMPUTE_EXPECT(dst.info()->is_resizable(), framework::LogLevel::ERRORS); ARM_COMPUTE_EXPECT(_target_indices.info()->is_resizable(), framework::LogLevel::ERRORS); - // TODO: uncomment after COMPMID-4363 - // add_padding_x({ &src, &dst, &_target_indices }, data_layout); + add_padding_x({ &src, &dst, &_target_indices }, data_layout); // Allocate tensors src.allocator()->allocate(); -- cgit v1.2.1