diff options
author | Declan-ARM <decmce01@arm.com> | 2024-02-07 13:07:31 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-02-09 14:01:11 +0000 |
commit | 5e90b831fc31605b5734531387837123fb31d0e0 (patch) | |
tree | c75cedfaa5f6170262eb3dfcea6e0522d89e3433 /src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp | |
parent | ddbda6a1ed13f7bef7e0dce07a37e91b062ce98a (diff) | |
download | armnn-5e90b831fc31605b5734531387837123fb31d0e0.tar.gz |
IVGCVSW-7569 GpuFsa Op: Add Reshape Operator
* Add Reshape EndToEnd tests to all backends
Signed-off-by: Declan-ARM <decmce01@arm.com>
Change-Id: Ic6d07ba8de0cf3271ed0e4c6d604e070ccb968e3
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp')
-rw-r--r-- | src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp index cf465c28ff..dce98b389f 100644 --- a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp @@ -176,6 +176,63 @@ TEST_CASE("IsLayerSupportedGpuFsaPooling2d") CHECK(supported); } +TEST_CASE("UNSUPPORTED_IsLayerSupportedGpuFsaReshape") +{ + TensorInfo inputInfo{}; + TensorInfo outputInfo{}; + + SUBCASE("Float32") + { + inputInfo = { { 2, 3 }, DataType::Float32 }; + outputInfo = { { 6 } , DataType::Float32 }; + } + + SUBCASE("Float16") + { + inputInfo = { { 2, 3 }, DataType::Float16 }; + outputInfo = { { 6 } , DataType::Float16 }; + } + + SUBCASE("Int32") + { + inputInfo = { { 2, 3 }, DataType::Signed32 }; + outputInfo = { { 6 } , DataType::Signed32 }; + } + + SUBCASE("Int16") + { + inputInfo = { { 2, 3 }, DataType::QSymmS16 }; + outputInfo = { { 6 } , DataType::QSymmS16 }; + } + + SUBCASE("UInt8") + { + inputInfo = { { 2, 3 }, DataType::QAsymmU8 }; + outputInfo = { { 6 } , DataType::QAsymmU8 }; + } + + SUBCASE("Int8") + { + inputInfo = { { 2, 3 }, DataType::QAsymmS8 }; + outputInfo = { { 6 } , DataType::QAsymmS8 }; + } + + ReshapeDescriptor desc; + desc.m_TargetShape = outputInfo.GetShape(); + + GpuFsaLayerSupport supportChecker; + std::string reasonIfNotSupported; + + auto supported = supportChecker.IsLayerSupported(LayerType::Reshape, + { inputInfo, outputInfo }, + desc, + EmptyOptional(), + EmptyOptional(), + reasonIfNotSupported); + + CHECK(!supported); +} + TEST_CASE("IsLayerSupportedGpuFsaResize") { TensorInfo inputInfo({ 1, 5, 5, 1 }, DataType::Float32); |