aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClWorkloadUtils.hpp
diff options
context:
space:
mode:
authorKeith 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
commit899f64f711fdac95d88795f4e88f7770c0cb94be (patch)
treebf33be2312ce4540674706b310a05a2b567e11cb /src/backends/cl/workloads/ClWorkloadUtils.hpp
parente9323ec3d8c638462e7269ecbef7d845b6a8b11a (diff)
downloadarmnn-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.hpp3
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;