diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2017-06-22 12:57:51 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | bbd9fb95daa08d6da67c567b40ca2cd032f7a2d3 (patch) | |
| tree | c1401585f64396d6f22bb790442d8183f3a17a9e /tests/dataset/PoolingLayerDataset.h | |
| parent | 2eac5bd444d16e4e81c427d5a99e1534b387e211 (diff) | |
| download | ComputeLibrary-bbd9fb95daa08d6da67c567b40ca2cd032f7a2d3.tar.gz | |
COMPMID-412: Port PoolingLayer to use fixed point 16.
Change-Id: I2005de4c7c14526996309826d33a0ec8e732d2d5
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/78720
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Steven Niu <steven.niu@arm.com>
Diffstat (limited to 'tests/dataset/PoolingLayerDataset.h')
| -rw-r--r-- | tests/dataset/PoolingLayerDataset.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/dataset/PoolingLayerDataset.h b/tests/dataset/PoolingLayerDataset.h index 5cdece4f66..1496cad379 100644 --- a/tests/dataset/PoolingLayerDataset.h +++ b/tests/dataset/PoolingLayerDataset.h @@ -134,7 +134,7 @@ public: ~GoogLeNetPoolingLayerDataset() = default; }; -class RandomPoolingLayerDataset final : public PoolingLayerDataset<8> +class RandomPoolingLayerDataset final : public PoolingLayerDataset<10> { public: RandomPoolingLayerDataset() @@ -148,6 +148,8 @@ public: PoolingLayerDataObject{ TensorShape(13U, 13U, 32U), TensorShape(6U, 6U, 32U), PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(2, 2, 0, 0)) }, PoolingLayerDataObject{ TensorShape(24U, 24U, 10U), TensorShape(12U, 12U, 10U), PoolingLayerInfo(PoolingType::AVG, 2, PadStrideInfo(2, 2, 0, 0)) }, PoolingLayerDataObject{ TensorShape(8U, 8U, 30U), TensorShape(4U, 4U, 30U), PoolingLayerInfo(PoolingType::AVG, 2, PadStrideInfo(2, 2, 0, 0)) }, + PoolingLayerDataObject{ TensorShape(7U, 7U, 10U), TensorShape(7U, 7U, 10U), PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(1, 1, 1, 1)) }, + PoolingLayerDataObject{ TensorShape(7U, 7U, 10U), TensorShape(7U, 7U, 10U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(1, 1, 1, 1)) }, } { } |