diff options
Diffstat (limited to 'tests/benchmark/fixtures')
-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); |