From 4cad860254f4249bd13f0fb43dfa04e1c84a8c91 Mon Sep 17 00:00:00 2001
From: Kevin May <kevin.may@arm.com>
Date: Tue, 18 May 2021 09:57:43 +0100
Subject: IVGCVSW-6026 Delegate VisitUnpackOperator not supported error

* Determine Splitter outputs to correctly perform validation
* Add validation for a Reshape Layer
* Add back the Num6 Axis2 test to CpuAcc and GpuAcc

Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I1e8108ce50d81420057d7a8b098a07eda63c5c8d
---
 delegate/src/test/UnpackTest.cpp | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'delegate/src/test')

diff --git a/delegate/src/test/UnpackTest.cpp b/delegate/src/test/UnpackTest.cpp
index e9ab6f73b9..c036f649ef 100644
--- a/delegate/src/test/UnpackTest.cpp
+++ b/delegate/src/test/UnpackTest.cpp
@@ -122,6 +122,12 @@ std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
 UnpackAxis0Num4Test<float>(tflite::TensorType_FLOAT32, backends);
 }
 
+TEST_CASE ("Unpack_Fp32_Axis2_Num6_CpuAcc_Test")
+{
+std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
+UnpackAxis2Num6Test<float>(tflite::TensorType_FLOAT32, backends);
+}
+
 // Uint8
 TEST_CASE ("Unpack_Uint8_Axis0_Num4_CpuAcc_Test")
 {
@@ -129,6 +135,12 @@ std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
 UnpackAxis0Num4Test<uint8_t>(tflite::TensorType_UINT8, backends);
 }
 
+TEST_CASE ("Unpack_Uint8_Axis2_Num6_CpuAcc_Test")
+{
+std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
+UnpackAxis2Num6Test<uint8_t>(tflite::TensorType_UINT8, backends);
+}
+
 } // End of Unpack_CpuAccTests
 
 TEST_SUITE("Unpack_GpuAccTests")
@@ -141,6 +153,12 @@ std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
 UnpackAxis0Num4Test<float>(tflite::TensorType_FLOAT32, backends);
 }
 
+TEST_CASE ("Unpack_Fp32_Axis2_Num6_GpuAcc_Test")
+{
+std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
+UnpackAxis2Num6Test<float>(tflite::TensorType_FLOAT32, backends);
+}
+
 // Uint8
 TEST_CASE ("Unpack_Uint8_Axis0_Num4_GpuAcc_Test")
 {
@@ -148,6 +166,12 @@ std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
 UnpackAxis0Num4Test<uint8_t>(tflite::TensorType_UINT8, backends);
 }
 
+TEST_CASE ("Unpack_Uint8_Axis2_Num6_GpuAcc_Test")
+{
+std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
+UnpackAxis2Num6Test<uint8_t>(tflite::TensorType_UINT8, backends);
+}
+
 } // End of Unpack_GpuAccTests
 
 // End of Unpack Test Suite
-- 
cgit v1.2.1