diff options
Diffstat (limited to 'delegate/src/test')
-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 |