diff options
Diffstat (limited to 'tests/InferenceTest.inl')
-rw-r--r-- | tests/InferenceTest.inl | 11 |
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; |