COMPMID-1937: Adds support for DequantizationLayer for NEON/CL.
Change-Id: I4b73edd176a277294e0e42e642460bc61210778a
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/744
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
diff --git a/tests/benchmark/NEON/DequantizationLayer.cpp b/tests/benchmark/NEON/DequantizationLayer.cpp
index 9a0a1e7..2ffa8a1 100644
--- a/tests/benchmark/NEON/DequantizationLayer.cpp
+++ b/tests/benchmark/NEON/DequantizationLayer.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017 ARM Limited.
+ * Copyright (c) 2017-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -41,8 +41,12 @@
{
namespace
{
-const auto data_types_src = framework::dataset::make("DataType", { DataType::U8 });
+const auto data_types_src = framework::dataset::make("DataType", { DataType::QASYMM8 });
+#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
+const auto data_types_dst = framework::dataset::make("DataType", { DataType::F16, DataType::F32 });
+#else /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
const auto data_types_dst = framework::dataset::make("DataType", { DataType::F32 });
+#endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
} // namespace
using NEDequantizationLayerFixture = DequantizationLayerFixture<Tensor, NEDequantizationLayer, Accessor>;
@@ -53,7 +57,7 @@
framework::DatasetMode::ALL,
framework::dataset::combine(framework::dataset::combine(datasets::Small3DShapes(), data_types_src), data_types_dst));
-TEST_SUITE_END()
+TEST_SUITE_END() // NEON
} // namespace benchmark
} // namespace test
} // namespace arm_compute