1 files changed, 12 insertions, 2 deletions
diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp
index d7c10cb599..9d44354cdd 100644
--- a/src/armnnSerializer/test/SerializerTests.cpp
+++ b/src/armnnSerializer/test/SerializerTests.cpp
@@ -76,11 +76,11 @@ BOOST_AUTO_TEST_CASE(SerializeAddition)
     deserializedNetwork->ExecuteStrategy(verifier);
 }
 
-BOOST_AUTO_TEST_CASE(SerializeArgMinMax)
+void SerializeArgMinMaxTest(armnn::DataType dataType)
 {
     const std::string layerName("argminmax");
     const armnn::TensorInfo inputInfo({1, 2, 3}, armnn::DataType::Float32);
-    const armnn::TensorInfo outputInfo({1, 3}, armnn::DataType::Signed32);
+    const armnn::TensorInfo outputInfo({1, 3}, dataType);
 
     armnn::ArgMinMaxDescriptor descriptor;
     descriptor.m_Function = armnn::ArgMinMaxFunction::Max;
@@ -107,6 +107,16 @@ BOOST_AUTO_TEST_CASE(SerializeArgMinMax)
     deserializedNetwork->ExecuteStrategy(verifier);
 }
 
+BOOST_AUTO_TEST_CASE(SerializeArgMinMaxSigned32)
+{
+    SerializeArgMinMaxTest(armnn::DataType::Signed32);
+}
+
+BOOST_AUTO_TEST_CASE(SerializeArgMinMaxSigned64)
+{
+    SerializeArgMinMaxTest(armnn::DataType::Signed64);
+}
+
 BOOST_AUTO_TEST_CASE(SerializeBatchNormalization)
 {
     const std::string layerName("batchNormalization");