diff options
Diffstat (limited to 'tests/datasets/GatherDataset.h')
-rw-r--r-- | tests/datasets/GatherDataset.h | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tests/datasets/GatherDataset.h b/tests/datasets/GatherDataset.h index 29a99d5239..f4ad7c9497 100644 --- a/tests/datasets/GatherDataset.h +++ b/tests/datasets/GatherDataset.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2019 Arm Limited. + * Copyright (c) 2018-2019, 2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -106,6 +106,18 @@ private: std::vector<int> _axis{}; }; + +class SmallGatherMultiDimIndicesDataset final : public GatherDataset +{ +public: + SmallGatherMultiDimIndicesDataset() + { + add_config(TensorShape(15U, 15U), TensorShape(4U, 13U, 2U), 0); + add_config(TensorShape(15U, 15U), TensorShape(2U, 12U), 0); + } +}; + + class SmallGatherDataset final : public GatherDataset { public: |