diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-03-01 16:03:50 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:50:48 +0000 |
commit | 57dac8400d56a4b68975d5563a9540c96d49fe5f (patch) | |
tree | cf474c6690c02115e623d5e7d867be665050f87e /tests/benchmark/fixtures/PoolingLayerFixture.h | |
parent | 0ef7e670a22248806458d7327db9e8b8c4db4ce6 (diff) | |
download | ComputeLibrary-57dac8400d56a4b68975d5563a9540c96d49fe5f.tar.gz |
COMPMID-806 Add NHWC data format support format for NEON pooling
Change-Id: I7ab174c72f3d56134fcec259a137739061fd12e9
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/123065
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/benchmark/fixtures/PoolingLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/PoolingLayerFixture.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/benchmark/fixtures/PoolingLayerFixture.h b/tests/benchmark/fixtures/PoolingLayerFixture.h index a7deb75370..bd7c155348 100644 --- a/tests/benchmark/fixtures/PoolingLayerFixture.h +++ b/tests/benchmark/fixtures/PoolingLayerFixture.h @@ -42,7 +42,7 @@ class PoolingLayerFixture : public framework::Fixture { public: template <typename...> - void setup(TensorShape src_shape, TensorShape dst_shape, PoolingLayerInfo info, DataType data_type, int batches) + void setup(TensorShape src_shape, TensorShape dst_shape, PoolingLayerInfo info, DataType data_type, DataLayout data_layout, int batches) { // Set batched in source and destination shapes const unsigned int fixed_point_position = 4; @@ -50,8 +50,8 @@ public: dst_shape.set(dst_shape.num_dimensions(), batches); // Create tensors - src = create_tensor<TensorType>(src_shape, data_type, 1, fixed_point_position); - dst = create_tensor<TensorType>(dst_shape, data_type, 1, fixed_point_position); + src = create_tensor<TensorType>(src_shape, data_type, 1, fixed_point_position, QuantizationInfo(), data_layout); + dst = create_tensor<TensorType>(dst_shape, data_type, 1, fixed_point_position, QuantizationInfo(), data_layout); // Create and configure function pool_layer.configure(&src, &dst, info); |