commit be88a57579a9a848efe13e6c524b5b104b871733
author Sadik Armagan <sadik.armagan@arm.com> Thu Apr 30 11:39:37 2020 +0100
committer Kevin May <kevin.may@arm.com> Thu Apr 30 10:46:12 2020 +0000
tree d5dc03627048f8ecd2d728b154434244f05475ea
parent 9937f9359ac4eeefc3535b66eddddd1b4f067c54

IVGCVSW-4753 Fix CpuAcc Hal 1.3 Softmax Failures

* Refactor Neon Softmax workload to accept supported data types

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I54aa72d5cbb862cafcc1eabe48f6a00d61050cd7

src/backends/reference/test/RefCreateWorkloadTests.cpp

diff --git a/src/backends/reference/test/RefCreateWorkloadTests.cpp b/src/backends/reference/test/RefCreateWorkloadTests.cpp
index 29bfbc0..4a57df7 100644
--- a/src/backends/reference/test/RefCreateWorkloadTests.cpp
+++ b/src/backends/reference/test/RefCreateWorkloadTests.cpp
@@ -504,10 +504,22 @@
     auto workload = CreateSoftmaxWorkloadTest<SoftmaxWorkloadType, DataType>(factory, graph);
 
     // Checks that outputs and inputs are as we expect them (see definition of CreateSoftmaxWorkloadTest).
+
+    armnn::TensorInfo tensorInfo({4, 1}, DataType);
+    if (DataType == armnn::DataType::QAsymmU8)
+    {
+        tensorInfo.SetQuantizationOffset(0);
+        tensorInfo.SetQuantizationScale(1.f / 256);
+    }
+    else if (DataType == armnn::DataType::QAsymmS8)
+    {
+        tensorInfo.SetQuantizationOffset(-128);
+        tensorInfo.SetQuantizationScale(1.f / 256);
+    }
     CheckInputOutput(
         std::move(workload),
-        TensorInfo({4, 1}, DataType),
-        TensorInfo({4, 1}, DataType));
+        tensorInfo,
+        tensorInfo);
 }
 
 BOOST_AUTO_TEST_CASE(CreateSoftmaxFloat32Workload)