From b7f5d172ccdb1d884388dd6e0e54f74241afca67 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 8 May 2018 11:26:06 +0100 Subject: COMPMID-1115: Valgrind: asserts triggered in NEPoolingLayerKernel Change-Id: I39573d6fe213a273bebeab66724781b1a134a6f8 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/130357 Reviewed-by: Giorgio Arena Tested-by: Jenkins Reviewed-by: Anthony Barbier --- tests/benchmark/fixtures/PoolingLayerFixture.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'tests/benchmark/fixtures') diff --git a/tests/benchmark/fixtures/PoolingLayerFixture.h b/tests/benchmark/fixtures/PoolingLayerFixture.h index a8263fbcdc..5a1a29612c 100644 --- a/tests/benchmark/fixtures/PoolingLayerFixture.h +++ b/tests/benchmark/fixtures/PoolingLayerFixture.h @@ -50,6 +50,12 @@ public: // Set batched in source and destination shapes const unsigned int fixed_point_position = 4; + // Permute shape if NHWC format + if(data_layout == DataLayout::NHWC) + { + permute(src_shape, PermutationVector(2U, 0U, 1U)); + } + TensorInfo src_info(src_shape, 1, data_type, fixed_point_position); src_info.set_data_layout(data_layout); -- cgit v1.2.1