aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/layerTests/StackTestImpl.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/StackTestImpl.cpp')
-rw-r--r--src/backends/backendsCommon/test/layerTests/StackTestImpl.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/StackTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/StackTestImpl.cpp
index 252adb93ac..6911c868ad 100644
--- a/src/backends/backendsCommon/test/layerTests/StackTestImpl.cpp
+++ b/src/backends/backendsCommon/test/layerTests/StackTestImpl.cpp
@@ -633,4 +633,12 @@ LayerTestResult<armnn::Half, 4> StackFloat16Test(
inputData,
outputExpectedData
);
+}
+
+LayerTestResult<int32_t, 4> StackInt32Test(
+ armnn::IWorkloadFactory& workloadFactory,
+ const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
+ const armnn::ITensorHandleFactory& tensorHandleFactory)
+{
+ return StackAxis0TestImpl<armnn::DataType::Signed32>(workloadFactory, memoryManager, tensorHandleFactory);
} \ No newline at end of file