diff options
author | Keith Davis <keith.davis@arm.com> | 2020-02-14 12:22:40 +0000 |
---|---|---|
committer | Keith Davis Arm <keith.davis@arm.com> | 2020-02-14 12:25:39 +0000 |
commit | a856501ac5c2d0cca70068993e5c7cc714872890 (patch) | |
tree | 45744f39e8852e5f3086d5aeeb333cfab8281133 /src/backends/backendsCommon | |
parent | 032bc74ca4bc6589a33f23db31accddc5b20baaa (diff) | |
download | armnn-a856501ac5c2d0cca70068993e5c7cc714872890.tar.gz |
IVGCVSW-4420 Failing Unit Test Cl: Quantize layer
* Fix bug with missing ValidateInputsOutputs in Cl Workload
* Renamed data type tests to be more specific
* Future proofing
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I157acc318e40d727fcfb3f36f7f577fc87472f3d
Diffstat (limited to 'src/backends/backendsCommon')
-rw-r--r-- | src/backends/backendsCommon/MakeWorkloadHelper.hpp | 1 | ||||
-rw-r--r-- | src/backends/backendsCommon/WorkloadFactory.cpp | 1 | ||||
-rw-r--r-- | src/backends/backendsCommon/WorkloadUtils.cpp | 1 |
3 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/MakeWorkloadHelper.hpp b/src/backends/backendsCommon/MakeWorkloadHelper.hpp index f876c6b781..250a10a61c 100644 --- a/src/backends/backendsCommon/MakeWorkloadHelper.hpp +++ b/src/backends/backendsCommon/MakeWorkloadHelper.hpp @@ -59,6 +59,7 @@ std::unique_ptr<IWorkload> MakeWorkloadHelper(const QueueDescriptorType& descrip case DataType::QAsymmU8: return MakeWorkloadForType<Uint8Workload>::Func(descriptor, info, std::forward<Args>(args)...); case DataType::QSymmS8: + case DataType::QAsymmS8: return MakeWorkloadForType<Int8Workload>::Func(descriptor, info, std::forward<Args>(args)...); case DataType::Signed32: return MakeWorkloadForType<Int32Workload>::Func(descriptor, info, std::forward<Args>(args)...); diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp index 283e748038..23ff70a52e 100644 --- a/src/backends/backendsCommon/WorkloadFactory.cpp +++ b/src/backends/backendsCommon/WorkloadFactory.cpp @@ -352,6 +352,7 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId, break; } case DataType::QAsymmU8: + case DataType::QAsymmS8: case DataType::QSymmS8: case DataType::QSymmS16: { diff --git a/src/backends/backendsCommon/WorkloadUtils.cpp b/src/backends/backendsCommon/WorkloadUtils.cpp index 69a62914e5..3b3959ba9f 100644 --- a/src/backends/backendsCommon/WorkloadUtils.cpp +++ b/src/backends/backendsCommon/WorkloadUtils.cpp @@ -166,6 +166,7 @@ armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstCpuTensorHandle* weightPermuted = ReorderWeightChannelsForAcl<half_float::half>(weightPermuted, dataLayout, permuteBuffer); break; + case DataType::QAsymmS8: case DataType::QAsymmU8: weightPermuted = ReorderWeightChannelsForAcl<uint8_t>(weightPermuted, dataLayout, permuteBuffer); break; |