diff options
| author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-01-08 11:33:44 +0000 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-01-14 10:55:58 +0000 |
| commit | 9c700378f2227cb9d51455ed4a5086daaac5532a (patch) | |
| tree | 53eb4acf5a9226e941e332b93db8ef260fb2d42b /tests/validation/fixtures/FullyConnectedLayerFixture.h | |
| parent | ab709a0ea0f7f5c8e02c315afffc300e09c783a8 (diff) | |
| download | ComputeLibrary-9c700378f2227cb9d51455ed4a5086daaac5532a.tar.gz | |
COMPMID-2769: Add support for QASYMM8_SIGNED in NEFullyConnectedLayer
Change-Id: I4c35c522375ae5a5de78716e079ebb9ffad15956
Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2581
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/fixtures/FullyConnectedLayerFixture.h')
| -rw-r--r-- | tests/validation/fixtures/FullyConnectedLayerFixture.h | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/tests/validation/fixtures/FullyConnectedLayerFixture.h b/tests/validation/fixtures/FullyConnectedLayerFixture.h index 0449d80de8..ff6ac17744 100644 --- a/tests/validation/fixtures/FullyConnectedLayerFixture.h +++ b/tests/validation/fixtures/FullyConnectedLayerFixture.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2019 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -49,7 +49,7 @@ template <typename TensorType, typename AccessorType, typename FunctionType, typ class FullyConnectedLayerValidationGenericFixture : public framework::Fixture { public: - using TBias = typename std::conditional<std::is_same<typename std::decay<T>::type, uint8_t>::value, int32_t, T>::type; + using TBias = typename std::conditional < std::is_same<typename std::decay<T>::type, uint8_t>::value || std::is_same<typename std::decay<T>::type, int8_t>::value, int32_t, T >::type; public: template <typename...> @@ -71,11 +71,16 @@ protected: template <typename U> void fill(U &&tensor, int i) { - if(is_data_type_quantized_asymmetric(_data_type)) + if(_data_type == DataType::QASYMM8) { std::uniform_int_distribution<uint8_t> distribution(0, 30); library->fill(tensor, distribution, i); } + else if(_data_type == DataType::QASYMM8_SIGNED) + { + std::uniform_int_distribution<int8_t> distribution(-15, 15); + library->fill(tensor, distribution, i); + } else if(_data_type == DataType::S32) { std::uniform_int_distribution<int32_t> distribution(-50, 50); |