diff options
author | Kevin May <kevin.may@arm.com> | 2021-05-18 09:57:43 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-05-18 12:42:30 +0000 |
commit | 4cad860254f4249bd13f0fb43dfa04e1c84a8c91 (patch) | |
tree | af894ec5a294e319b1d34b6054bac6dbf17409c3 /delegate/src/test/UnpackTest.cpp | |
parent | 25ab3a8326a9e2c52c84b2747fa72907109a695d (diff) | |
download | armnn-4cad860254f4249bd13f0fb43dfa04e1c84a8c91.tar.gz |
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
Diffstat (limited to 'delegate/src/test/UnpackTest.cpp')
-rw-r--r-- | delegate/src/test/UnpackTest.cpp | 24 |
1 files changed, 24 insertions, 0 deletions
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 |