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);