diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-06-24 12:43:38 +0100 |
---|---|---|
committer | Ferran Balaguer Arm <ferran.balaguer@arm.com> | 2019-06-26 15:22:31 +0000 |
commit | b2b5a26e7b7250532df189cd7cf7cf1d4b528a28 (patch) | |
tree | 6cbb1d44f074455a58ab00c2f6edd5ab5c3e8ebe /src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp | |
parent | 5191033c76330509666e0993857b7286e2a325fc (diff) | |
download | armnn-b2b5a26e7b7250532df189cd7cf7cf1d4b528a28.tar.gz |
Update the CL pin to the latest master
* Update SoftMaxUint8Workload for CL and NEON to deal with
quantization parameters as vectors.
* Change Sigmoid Activation function QAsymm8 tests to use
scale=1.f/256.f and offset=0 as quatization output parameters.
!android-nn-driver:1417
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: Ief91f10193fbbbad0c0124ece41f0bf4e0dcd992
Diffstat (limited to 'src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp b/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp index 086f3751b4..84d735cdf7 100644 --- a/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp +++ b/src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp @@ -25,7 +25,10 @@ ClSoftmaxUint8Workload::ClSoftmaxUint8Workload(const SoftmaxQueueDescriptor& des const auto outputQuantization = output.info()->quantization_info(); - if ((outputQuantization.scale != (1.0f / 256.0f)) || (outputQuantization.offset != 0)) + if (((!outputQuantization.scale.empty()) && (outputQuantization.scale[0] != (1.0f / 256.0f))) || + ((!outputQuantization.offset.empty()) && (outputQuantization.offset[0] != 0)) || + (outputQuantization.scale.empty()) || + (outputQuantization.offset.empty())) { throw InvalidArgumentException( "Invalid quantization for output. Only scale = 1.0f / 256.0f and offset = 0 supported"); |