diff options
| author | SiCongLi <sicong.li@arm.com> | 2020-03-02 15:39:15 +0000 |
|---|---|---|
| committer | SiCong Li <sicong.li@arm.com> | 2020-03-25 11:23:19 +0000 |
| commit | 2e5fd637205770ec5e11096e6e19b8efc67d544e (patch) | |
| tree | c7f5b53d6ec54cf39fcc638bad9d054c3e04bf43 /tests/datasets/FullyConnectedLayerDataset.h | |
| parent | 51a9558d484e812707a776efceef18b93698cec9 (diff) | |
| download | ComputeLibrary-2e5fd637205770ec5e11096e6e19b8efc67d544e.tar.gz | |
COMPMID-3098 Fuse Relu and Bounded Relu with FullyConnected NEON
Change-Id: Id28062445590d6c06b35f7d7434eb38393ae94a7
Signed-off-by: SiCongLi <sicong.li@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2875
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets/FullyConnectedLayerDataset.h')
| -rw-r--r-- | tests/datasets/FullyConnectedLayerDataset.h | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/tests/datasets/FullyConnectedLayerDataset.h b/tests/datasets/FullyConnectedLayerDataset.h index 085e9c76b6..06f74ec588 100644 --- a/tests/datasets/FullyConnectedLayerDataset.h +++ b/tests/datasets/FullyConnectedLayerDataset.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -114,6 +114,17 @@ private: std::vector<TensorShape> _dst_shapes{}; }; +class FullyConnectedLayerWithActivationDataset final : public FullyConnectedLayerDataset +{ +public: + FullyConnectedLayerWithActivationDataset() + { + // Conv -> FC + add_config(TensorShape(8U, 1U, 1U), TensorShape(8U, 16U), TensorShape(16U), TensorShape(16U)); + // FC -> FC + add_config(TensorShape(1U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U)); + } +}; class TinyFullyConnectedLayerDataset final : public FullyConnectedLayerDataset { public: |