diff options
| author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | 2024-03-18 12:13:30 +0000 |
|---|---|---|
| committer | Suhail M <MohammedSuhail.Munshi@arm.com> | 2024-03-25 13:23:33 +0000 |
| commit | 473b8291a7dc126586d19b82d6c84b4c7a6e44a6 (patch) | |
| tree | 1ca111dfac74ddfb1d8430b6503ba7e6402e9c5d /tests/datasets | |
| parent | 4908981a4db9da7ec51fd0c2900a25bd91b2926a (diff) | |
| download | ComputeLibrary-473b8291a7dc126586d19b82d6c84b4c7a6e44a6.tar.gz | |
Adds Tests and reference implementation for scatter operator with 1D tensors.
Resolves: [COMPMID-6890]
Change-Id: Ie4a8db24fc6387afa9ddf42b3607e040cdf8df67
Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11339
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets')
| -rw-r--r-- | tests/datasets/ScatterDataset.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/datasets/ScatterDataset.h b/tests/datasets/ScatterDataset.h index 09f6338432..d204d17855 100644 --- a/tests/datasets/ScatterDataset.h +++ b/tests/datasets/ScatterDataset.h @@ -113,12 +113,13 @@ private: std::vector<TensorShape> _dst_shapes{}; }; -class SmallScatterDataset final : public ScatterDataset +class Small1DScatterDataset final : public ScatterDataset { public: - SmallScatterDataset() + Small1DScatterDataset() { add_config(TensorShape(6U), TensorShape(6U), TensorShape(6U), TensorShape(6U)); + add_config(TensorShape(10U), TensorShape(2U), TensorShape(2U), TensorShape(10U)); } }; } // namespace datasets |