From 4408850bd426481e39ae9182aff4d563b6836181 Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Mon, 27 Jul 2020 11:27:19 +0100
Subject: IVGCVSW-4712 Add EndToEnd test for FILL

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ic89bcbbd580abe1b05bd26748db704e83cf65bea
---
 src/backends/reference/test/RefEndToEndTests.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'src/backends/reference/test')

diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp
index 45605e0fc6..88876877bf 100644
--- a/src/backends/reference/test/RefEndToEndTests.cpp
+++ b/src/backends/reference/test/RefEndToEndTests.cpp
@@ -14,6 +14,7 @@
 #include <backendsCommon/test/DequantizeEndToEndTestImpl.hpp>
 #include <backendsCommon/test/DetectionPostProcessEndToEndTestImpl.hpp>
 #include <backendsCommon/test/ElementwiseUnaryEndToEndTestImpl.hpp>
+#include <backendsCommon/test/FillEndToEndTestImpl.hpp>
 #include <backendsCommon/test/GatherEndToEndTestImpl.hpp>
 #include <backendsCommon/test/InstanceNormalizationEndToEndTestImpl.hpp>
 #include <backendsCommon/test/LogSoftmaxEndToEndTestImpl.hpp>
@@ -582,6 +583,21 @@ BOOST_AUTO_TEST_CASE(RefEluEndToEndTestQSymmS16)
     EluEndToEndTest<armnn::DataType::QSymmS16>(defaultBackends);
 }
 
+BOOST_AUTO_TEST_CASE(RefFillEndToEndTest)
+{
+    FillEndToEnd<armnn::DataType::Float32>(defaultBackends);
+}
+
+//BOOST_AUTO_TEST_CASE(RefFillEndToEndTestFloat16)
+//{
+//    FillEndToEnd<armnn::DataType::Float16>(defaultBackends);
+//}
+
+BOOST_AUTO_TEST_CASE(RefFillEndToEndTestInt32)
+{
+    FillEndToEnd<armnn::DataType::Signed32>(defaultBackends);
+}
+
 BOOST_AUTO_TEST_CASE(RefGatherFloatTest)
 {
     GatherEndToEnd<armnn::DataType::Float32>(defaultBackends);
-- 
cgit v1.2.1