diff options
author | Keith Davis <keith.davis@arm.com> | 2019-11-26 16:01:18 +0000 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-11-27 10:07:47 +0000 |
commit | 899f64f711fdac95d88795f4e88f7770c0cb94be (patch) | |
tree | bf33be2312ce4540674706b310a05a2b567e11cb /src/backends/cl/workloads/ClWorkloadUtils.hpp | |
parent | e9323ec3d8c638462e7269ecbef7d845b6a8b11a (diff) | |
download | armnn-899f64f711fdac95d88795f4e88f7770c0cb94be.tar.gz |
IVGCVSW-3859 Per-channel QSymm8 support of Convolution2d in Cl
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: Ie43f20e6fb2f9b4a1e9e9eb561fcf7b5d6a01806
Diffstat (limited to 'src/backends/cl/workloads/ClWorkloadUtils.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClWorkloadUtils.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClWorkloadUtils.hpp b/src/backends/cl/workloads/ClWorkloadUtils.hpp index f5e60e6315..ecaf2799e8 100644 --- a/src/backends/cl/workloads/ClWorkloadUtils.hpp +++ b/src/backends/cl/workloads/ClWorkloadUtils.hpp @@ -101,6 +101,9 @@ inline void InitializeArmComputeClTensorData(arm_compute::CLTensor& clTensor, case DataType::QuantisedAsymm8: CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<uint8_t>()); break; + case DataType::QuantizedSymm8PerAxis: + CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int8_t>()); + break; case DataType::Signed32: CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int32_t>()); break; |