aboutsummaryrefslogtreecommitdiff
path: root/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp
diff options
context:
space:
mode:
authorDeclan-ARM <decmce01@arm.com>2024-02-07 13:07:31 +0000
committerTeresaARM <teresa.charlinreyes@arm.com>2024-02-09 14:01:11 +0000
commit5e90b831fc31605b5734531387837123fb31d0e0 (patch)
treec75cedfaa5f6170262eb3dfcea6e0522d89e3433 /src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp
parentddbda6a1ed13f7bef7e0dce07a37e91b062ce98a (diff)
downloadarmnn-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.cpp57
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);