diff options
Diffstat (limited to 'src/backends/backendsCommon/test/QuantizationEndToEndTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/QuantizationEndToEndTestImpl.hpp | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/QuantizationEndToEndTestImpl.hpp b/src/backends/backendsCommon/test/QuantizationEndToEndTestImpl.hpp index f5c2eea601..3039b9b5a3 100644 --- a/src/backends/backendsCommon/test/QuantizationEndToEndTestImpl.hpp +++ b/src/backends/backendsCommon/test/QuantizationEndToEndTestImpl.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2023-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -105,4 +105,24 @@ void QuantizationEndToEndFloat16(const std::vector<armnn::BackendId>& backends) qOffset); }; +inline void QuantizationEndToEndInt8(const std::vector<armnn::BackendId>& backends) +{ + using namespace armnn; + + const TensorShape tensorShape({ 1, 1, 1, 5 }); + + std::vector<int8_t> inputData = { 113, 16, 13, 101, 13 }; + std::vector<int8_t> expectedOutputData = { 127, 45, 41, 127, 41 }; + + float qScale = 0.75f; + int32_t qOffset = 24; + + QuantizeEndToEndLayerTestImpl<DataType::QSymmS8, DataType::QSymmS8>(backends, + tensorShape, + inputData, + expectedOutputData, + qScale, + qOffset); +}; + }
\ No newline at end of file |