From 20bea0071d507772e303eb6f1c476bf1feac9be5 Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Wed, 16 Oct 2019 09:29:38 +0100
Subject: IVGCVSW-3895 Add EndToEndLayerTest for the Reference Implementation
 of LOG_SOFTMAX

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I500f0b819be543dc5829990d363e5d911197c9ec
---
 src/backends/reference/test/RefEndToEndTests.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/backends/reference')

diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp
index 8b768abdee..370ef6599b 100644
--- a/src/backends/reference/test/RefEndToEndTests.cpp
+++ b/src/backends/reference/test/RefEndToEndTests.cpp
@@ -14,6 +14,7 @@
 #include <backendsCommon/test/DetectionPostProcessEndToEndTestImpl.hpp>
 #include <backendsCommon/test/GatherEndToEndTestImpl.hpp>
 #include <backendsCommon/test/InstanceNormalizationEndToEndTestImpl.hpp>
+#include <backendsCommon/test/LogSoftmaxEndToEndTestImpl.hpp>
 #include <backendsCommon/test/PreluEndToEndTestImpl.hpp>
 #include <backendsCommon/test/ResizeEndToEndTestImpl.hpp>
 #include <backendsCommon/test/SpaceToDepthEndToEndTestImpl.hpp>
@@ -778,6 +779,12 @@ BOOST_AUTO_TEST_CASE(RefDetectionPostProcessFastNmsUint8Test)
                                                                           1.0f, 1, 0.01f, 0, 0.5f, 0);
 }
 
+// LogSoftmax
+BOOST_AUTO_TEST_CASE(RefLogSoftmaxEndToEndTest)
+{
+    LogSoftmaxEndToEndTest(defaultBackends);
+}
+
 BOOST_AUTO_TEST_CASE(RefPreluEndToEndTestFloat32)
 {
     PreluEndToEndNegativeTest<armnn::DataType::Float32>(defaultBackends);
-- 
cgit v1.2.1