diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2018-01-31 13:05:09 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:45:00 +0000 |
| commit | 1c0d0ffb99814749d5c48df282dc212cb939094a (patch) | |
| tree | 9e7edd3d17d46c1d41a98b73ca1ba958d6282f02 /tests/datasets/FullyConnectedLayerDataset.h | |
| parent | cdd1e85f69a867de3958b55560cc3f00943ea9c7 (diff) | |
| download | ComputeLibrary-1c0d0ffb99814749d5c48df282dc212cb939094a.tar.gz | |
COMPMID-832: Decrease validation coverage of QS8/QS16
Change-Id: I5366d11aefdb8f3ba7326ed7527eb216c4de0668
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118372
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets/FullyConnectedLayerDataset.h')
| -rw-r--r-- | tests/datasets/FullyConnectedLayerDataset.h | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/datasets/FullyConnectedLayerDataset.h b/tests/datasets/FullyConnectedLayerDataset.h index 60183f8746..085e9c76b6 100644 --- a/tests/datasets/FullyConnectedLayerDataset.h +++ b/tests/datasets/FullyConnectedLayerDataset.h @@ -114,6 +114,21 @@ private: std::vector<TensorShape> _dst_shapes{}; }; +class TinyFullyConnectedLayerDataset final : public FullyConnectedLayerDataset +{ +public: + TinyFullyConnectedLayerDataset() + { + // Conv -> FC + add_config(TensorShape(8U, 1U, 1U), TensorShape(8U, 16U), TensorShape(16U), TensorShape(16U)); + // Conv -> FC + add_config(TensorShape(1U, 1U, 1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U)); + // FC -> FC + add_config(TensorShape(1U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U)); + // FC -> FC (batched) + add_config(TensorShape(1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U)); + } +}; class SmallFullyConnectedLayerDataset final : public FullyConnectedLayerDataset { public: |