Add and tidy up activation and elementwise binary end to end tests

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I9714c4c57e923ac775dcde2951de07cea35c40ee

1 files changed, 60 insertions, 22 deletions

diff --git a/src/backends/neon/test/NeonEndToEndTests.cpp b/src/backends/neon/test/NeonEndToEndTests.cpp
index 1bf9344883..f505b7fd46 100644
--- a/src/backends/neon/test/NeonEndToEndTests.cpp
+++ b/src/backends/neon/test/NeonEndToEndTests.cpp
@@ -161,23 +161,23 @@ TEST_CASE("NeonAdditionEndToEndFloat16Simple3DTest")
     ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float16>(neonDefaultBackends, BinaryOperation::Add);
 }
 
-// Power
-TEST_CASE("NeonPowerEndToEndTestFloat32")
+// Div
+TEST_CASE("NeonDivEndToEndTestFloat32")
 {
-    ElementwiseBinarySimpleEndToEnd<armnn::DataType::Float32>(neonDefaultBackends, BinaryOperation::Power);
+    ElementwiseBinarySimpleEndToEnd<armnn::DataType::Float32>(neonDefaultBackends, BinaryOperation::Div);
 }
 
-// SqDiff
-TEST_CASE("NeonSquaredDifferenceEndToEndTestFloat32")
+// Mul
+TEST_CASE("NeonMulEndToEndTestFloat32")
 {
-    ElementwiseBinarySimpleEndToEnd<armnn::DataType::Float32>(neonDefaultBackends, BinaryOperation::SqDiff);
+    ElementwiseBinarySimpleEndToEnd<armnn::DataType::Float32>(neonDefaultBackends, BinaryOperation::Mul);
 }
-
-TEST_CASE("NeonSquaredDifferenceEndToEndTestUint8")
+TEST_CASE("NeonMulEndToEndTestUint8")
 {
-    ElementwiseBinarySimpleEndToEnd<armnn::DataType::QAsymmU8>(neonDefaultBackends, BinaryOperation::SqDiff);
+    ElementwiseBinarySimpleEndToEnd<armnn::DataType::QAsymmU8>(neonDefaultBackends, BinaryOperation::Mul);
 }
 
+// Sub
 TEST_CASE("NeonSubtractionEndToEndFloat32Simple3DTest")
 {
     ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float32>(neonDefaultBackends, BinaryOperation::Sub);
@@ -187,6 +187,23 @@ TEST_CASE("NeonSubtractionEndToEndFloat16Simple3DTest")
     ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float16>(neonDefaultBackends, BinaryOperation::Sub);
 }
 
+// Power
+TEST_CASE("NeonPowerEndToEndTestFloat32")
+{
+    ElementwiseBinarySimpleEndToEnd<armnn::DataType::Float32>(neonDefaultBackends, BinaryOperation::Power);
+}
+
+// SqDiff
+TEST_CASE("NeonSquaredDifferenceEndToEndTestFloat32")
+{
+    ElementwiseBinarySimpleEndToEnd<armnn::DataType::Float32>(neonDefaultBackends, BinaryOperation::SqDiff);
+}
+
+TEST_CASE("NeonSquaredDifferenceEndToEndTestUint8")
+{
+    ElementwiseBinarySimpleEndToEnd<armnn::DataType::QAsymmU8>(neonDefaultBackends, BinaryOperation::SqDiff);
+}
+
 // Batch Mat Mul
 TEST_CASE("NeonBatchMatMulEndToEndFloat32Test")
 {
@@ -281,37 +298,56 @@ TEST_CASE("DequantizeEndToEndOffsetTest")
     DequantizeEndToEndOffset<armnn::DataType::QAsymmU8>(neonDefaultBackends);
 }
 
-TEST_CASE("NeonEluEndToEndTestFloat32")
+// Activations
+// Linear
+TEST_CASE("NeonLinearEndToEndTestFloat32")
 {
-    EluEndToEndTest<armnn::DataType::Float32>(neonDefaultBackends);
+    ActivationEndToEndTest<armnn::DataType::Float32>(neonDefaultBackends, ActivationFunction::Linear);
 }
 
-TEST_CASE("NeonEluEndToEndTestFloat16")
+// Sigmoid
+TEST_CASE("NeonSigmoidEndToEndTestFloat32")
 {
-    EluEndToEndTest<armnn::DataType::Float16>(neonDefaultBackends);
+    ActivationEndToEndTest<armnn::DataType::Float32>(neonDefaultBackends, ActivationFunction::Sigmoid);
 }
 
-// HardSwish
-TEST_CASE("NeonHardSwishEndToEndTestFloat32")
+// ReLu
+TEST_CASE("NeonReLuEndToEndTestFloat32")
 {
-    HardSwishEndToEndTest<armnn::DataType::Float32>(neonDefaultBackends);
+    ActivationEndToEndTest<armnn::DataType::Float32>(neonDefaultBackends, ActivationFunction::ReLu);
 }
 
-TEST_CASE("NeonHardSwishEndToEndTestFloat16")
+// BoundedReLu
+TEST_CASE("NeonBoundedReLuEndToEndTestFloat32")
 {
-    HardSwishEndToEndTest<armnn::DataType::Float16>(neonDefaultBackends);
+    ActivationEndToEndTest<armnn::DataType::Float32>(neonDefaultBackends, ActivationFunction::BoundedReLu);
 }
 
-TEST_CASE("NeonHardSwishEndToEndTestQAsymmS8")
+// LeakyRelu
+TEST_CASE("NeonLeakyReluActivationFloat32")
+{
+    ActivationEndToEndTest<DataType::Float32>(neonDefaultBackends, ActivationFunction::LeakyReLu, 1.f, 0,  0.01f);
+}
+
+// Elu
+TEST_CASE("NeonEluEndToEndTestFloat32")
+{
+    ActivationEndToEndTest<DataType::Float32>(neonDefaultBackends, ActivationFunction::Elu);
+}
+
+// HardSwish
+TEST_CASE("NeonHardSwishEndToEndTestFloat32")
 {
-    HardSwishEndToEndTest<armnn::DataType::QAsymmS8>(neonDefaultBackends);
+    ActivationEndToEndTest<DataType::Float32>(neonDefaultBackends, ActivationFunction::HardSwish);
 }
 
-TEST_CASE("NeonHardSwishEndToEndTestQAsymmU8")
+// TanH
+TEST_CASE("NeonTanHEndToEndTestFloat32")
 {
-    HardSwishEndToEndTest<armnn::DataType::QAsymmU8>(neonDefaultBackends);
+    ActivationEndToEndTest<DataType::Float32>(neonDefaultBackends, ActivationFunction::TanH, 1.f, 0, 2, 3);
 }
 
+// Prelu
 TEST_CASE("NeonPreluEndToEndFloat32Test")
 {
     PreluEndToEndNegativeTest<armnn::DataType::Float32>(neonDefaultBackends);
@@ -322,6 +358,7 @@ TEST_CASE("NeonPreluEndToEndTestUint8Test")
     PreluEndToEndPositiveTest<armnn::DataType::QAsymmU8>(neonDefaultBackends);
 }
 
+// SpaceToDepth
 TEST_CASE("NeonSpaceToDepthNhwcEndToEndTest1")
 {
     SpaceToDepthNhwcEndToEndTest1(neonDefaultBackends);
@@ -342,6 +379,7 @@ TEST_CASE("NeonSpaceToDepthNchwEndToEndTest2")
     SpaceToDepthNchwEndToEndTest2(neonDefaultBackends);
 }
 
+// Split
 TEST_CASE("NeonSplitter1dEndToEndTest")
 {
     Splitter1dEndToEnd<armnn::DataType::Float32>(neonDefaultBackends);