aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/ScatterDataset.h
diff options
context:
space:
mode:
authorGunes Bayir <gunes.bayir@arm.com>2024-04-24 10:27:13 +0100
committerGunes Bayir <gunes.bayir@arm.com>2024-04-25 15:41:36 +0000
commitada3200f5cec0b6a37f898d5d6f8e69395d7bcb1 (patch)
tree4e0b1e1e5c10a07307c1d2ac9a76c606849629b1 /tests/datasets/ScatterDataset.h
parent62d600fe8c0afba81cc5f5dd315eb6dcc04f90b8 (diff)
downloadComputeLibrary-ada3200f5cec0b6a37f898d5d6f8e69395d7bcb1.tar.gz
Add update/index/output (m+1)/2d/(m+n) support for CLScatter
Resolves: COMPMID-6894, COMPMID-6896 Change-Id: I9d29fd3701a7e0f28d83f81a6c42a7234c2587c3 Signed-off-by: Gunes Bayir <gunes.bayir@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11477 Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Ramy Elgammal <ramy.elgammal@arm.com> Dynamic-Fusion: Ramy Elgammal <ramy.elgammal@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/datasets/ScatterDataset.h')
-rw-r--r--tests/datasets/ScatterDataset.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/datasets/ScatterDataset.h b/tests/datasets/ScatterDataset.h
index 8b0972f99a..c0858941db 100644
--- a/tests/datasets/ScatterDataset.h
+++ b/tests/datasets/ScatterDataset.h
@@ -137,7 +137,7 @@ public:
// - src/updates/dst should all have same number of dims. Indices should be 2D.
add_config(TensorShape(6U, 5U), TensorShape(6U, 2U), TensorShape(1U, 2U), TensorShape(6U, 5U));
add_config(TensorShape(9U, 3U, 4U), TensorShape(9U, 3U, 2U), TensorShape(1U, 2U), TensorShape(9U, 3U, 4U));
- add_config(TensorShape(3U, 2U, 4U, 2U), TensorShape(3U, 2U, 4U, 2U), TensorShape(1U, 2U), TensorShape(3U, 2U, 4U, 2U));
+ add_config(TensorShape(17U, 3U, 2U, 4U), TensorShape(17U, 3U, 2U, 7U), TensorShape(1U, 7U), TensorShape(17U, 3U, 2U, 4U));
}
};