diff options
Diffstat (limited to 'tests/InferenceTest.inl')
-rw-r--r-- | tests/InferenceTest.inl | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl index b6087c5e5a..94dbfe78b8 100644 --- a/tests/InferenceTest.inl +++ b/tests/InferenceTest.inl @@ -67,6 +67,14 @@ struct ClassifierResultProcessor }); } + void operator()(const std::vector<armnn::Half>& values) + { + SortPredictions(values, [](armnn::Half value) + { + return value; + }); + } + void operator()(const std::vector<int8_t>& values) { SortPredictions(values, [](int8_t value) |