aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/CL/Select.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/CL/Select.cpp')
-rw-r--r--tests/validation/CL/Select.cpp35
1 files changed, 34 insertions, 1 deletions
diff --git a/tests/validation/CL/Select.cpp b/tests/validation/CL/Select.cpp
index 3d7c61aab5..d3540cae48 100644
--- a/tests/validation/CL/Select.cpp
+++ b/tests/validation/CL/Select.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018-2020 Arm Limited.
+ * Copyright (c) 2018-2021 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -107,6 +107,17 @@ FIXTURE_DATA_TEST_CASE(RunSmall,
validate(CLAccessor(_target), _reference);
}
+FIXTURE_DATA_TEST_CASE(RunOneDim,
+ CLSelectFixture<half>,
+ framework::DatasetMode::PRECOMMIT,
+ combine(combine(framework::dataset::make("Shape", TensorShape(1U, 16U)),
+ framework::dataset::make("has_same_rank", { false, true })),
+ framework::dataset::make("DataType", DataType::F16)))
+{
+ // Validate output
+ validate(CLAccessor(_target), _reference);
+}
+
FIXTURE_DATA_TEST_CASE(RunLarge,
CLSelectFixture<half>,
framework::DatasetMode::NIGHTLY,
@@ -127,6 +138,17 @@ FIXTURE_DATA_TEST_CASE(RunSmall,
validate(CLAccessor(_target), _reference);
}
+FIXTURE_DATA_TEST_CASE(RunOneDim,
+ CLSelectFixture<float>,
+ framework::DatasetMode::PRECOMMIT,
+ combine(combine(framework::dataset::make("Shape", TensorShape(1U, 16U)),
+ framework::dataset::make("has_same_rank", { false, true })),
+ framework::dataset::make("DataType", DataType::F32)))
+{
+ // Validate output
+ validate(CLAccessor(_target), _reference);
+}
+
FIXTURE_DATA_TEST_CASE(RunLarge,
CLSelectFixture<float>,
framework::DatasetMode::NIGHTLY,
@@ -149,6 +171,17 @@ FIXTURE_DATA_TEST_CASE(RunSmall,
validate(CLAccessor(_target), _reference);
}
+FIXTURE_DATA_TEST_CASE(RunOneDim,
+ CLSelectFixture<uint8_t>,
+ framework::DatasetMode::PRECOMMIT,
+ combine(combine(framework::dataset::make("Shape", TensorShape(1U, 16U)),
+ framework::dataset::make("has_same_rank", { false, true })),
+ framework::dataset::make("DataType", DataType::QASYMM8)))
+{
+ // Validate output
+ validate(CLAccessor(_target), _reference);
+}
+
FIXTURE_DATA_TEST_CASE(RunLarge,
CLSelectFixture<uint8_t>,
framework::DatasetMode::NIGHTLY,