1 files changed, 10 insertions, 1 deletions
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl
index 3d6dae335a..79700d991b 100644
--- a/tests/InferenceTest.inl
+++ b/tests/InferenceTest.inl
@@ -26,7 +26,8 @@ namespace armnn
 namespace test
 {
 
-using TContainer = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>>;
+using TContainer =
+        mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>, std::vector<int8_t>>;
 
 template <typename TTestCaseDatabase, typename TModel>
 ClassifierTestCase<TTestCaseDatabase, TModel>::ClassifierTestCase(
@@ -66,6 +67,14 @@ struct ClassifierResultProcessor
                                 });
     }
 
+    void operator()(const std::vector<int8_t>& values)
+    {
+        SortPredictions(values, [](int8_t value)
+        {
+            return value;
+        });
+    }
+
     void operator()(const std::vector<uint8_t>& values)
     {
         auto& scale = m_Scale;