aboutsummaryrefslogtreecommitdiff
path: root/tests/InferenceTest.inl
diff options
context:
space:
mode:
authorFinn Williams <Finn.Williams@arm.com>2021-02-22 15:13:12 +0000
committerJim Flynn <jim.flynn@arm.com>2021-03-03 17:06:43 +0000
commitf806c4d075814a9dc9d206a4db123d3060ad7ebd (patch)
treea110c106598a6830d0862526742314f5048b8acb /tests/InferenceTest.inl
parent82c59d75ef0f191887fae1cc2864bbf4b37ac0c5 (diff)
downloadarmnn-experimental/abi-tests.tar.gz
IVGCVSW-5612 Fix tiny_wav2letter_relu_fixed_int8 delegate outputexperimental/abi-tests
* fix delegate perchannel quantization * change delegate to check reshape options before inputs * Add int8 "qsymms8" option to ExecuteNetwork * Add option to run ExecuteNetwork on tflite w/o delegate !referencetests:301301 Signed-off-by: Finn Williams <Finn.Williams@arm.com> Change-Id: If3e12599b17aff1199d7ab0a55e1c901e480083d
Diffstat (limited to 'tests/InferenceTest.inl')
-rw-r--r--tests/InferenceTest.inl11
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;