diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-05-08 11:26:06 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:51:17 +0000 |
commit | b7f5d172ccdb1d884388dd6e0e54f74241afca67 (patch) | |
tree | 6ee797354a17a8e74e61bc51349713aa79c8496f /tests/benchmark/fixtures/PoolingLayerFixture.h | |
parent | 4497e8170793208aa0ccc409de72ad9f5c8a56ee (diff) | |
download | ComputeLibrary-b7f5d172ccdb1d884388dd6e0e54f74241afca67.tar.gz |
COMPMID-1115: Valgrind: asserts triggered in NEPoolingLayerKernel
Change-Id: I39573d6fe213a273bebeab66724781b1a134a6f8
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/130357
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/benchmark/fixtures/PoolingLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/PoolingLayerFixture.h | 6 |
1 files changed, 6 insertions, 0 deletions
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); |