From a799ce0ad775829862891dd98d1232638ec8761e Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 12 Sep 2018 20:11:34 +0100 Subject: COMPMID-1564: Add NEDepthwiseConvolution3x3 for QASYMM8 Change-Id: I1f55508af6f220e5f41df7b56daffb4761ed0591 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/148253 Tested-by: bsgcomp Reviewed-by: Isabella Gottardi --- utils/Utils.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'utils') diff --git a/utils/Utils.h b/utils/Utils.h index 0bbdcc25d1..130e1f72fe 100644 --- a/utils/Utils.h +++ b/utils/Utils.h @@ -357,7 +357,7 @@ public: void fill_tensor(T &tensor) { ARM_COMPUTE_ERROR_ON(!is_open()); - ARM_COMPUTE_ERROR_ON_DATA_TYPE_NOT_IN(&tensor, arm_compute::DataType::F32); + ARM_COMPUTE_ERROR_ON_DATA_TYPE_NOT_IN(&tensor, arm_compute::DataType::QASYMM8, arm_compute::DataType::S32, arm_compute::DataType::F32); try { // Map buffer if creating a CLTensor @@ -413,6 +413,8 @@ public: switch(tensor.info()->data_type()) { + case arm_compute::DataType::QASYMM8: + case arm_compute::DataType::S32: case arm_compute::DataType::F32: { // Read data -- cgit v1.2.1