commit e9323ec3d8c638462e7269ecbef7d845b6a8b11a
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Tue Nov 26 12:50:34 2019 +0000
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Wed Nov 27 09:53:32 2019 +0000
tree 930d60a90fe65ca44cab2456105e464c2f1abcf7
parent 94c4fef71594a60e4ea27aa6fa794c7ed3be4a27

Add IsQuantizeType(DataType) for use in validation functions

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ib790b24cfde87c92e29615da942c25bf7c98f95e

src/backends/backendsCommon/WorkloadData.cpp

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 83e823c..6d5c6e8 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -2483,10 +2483,7 @@
     const TensorInfo& inputTensorInfo  = workloadInfo.m_InputTensorInfos[0];
     const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0];
 
-    if (inputTensorInfo.GetDataType() != DataType::QuantisedAsymm8 &&
-        inputTensorInfo.GetDataType() != DataType::QuantisedSymm8 &&
-        inputTensorInfo.GetDataType() != DataType::QuantizedSymm8PerAxis &&
-        inputTensorInfo.GetDataType() != DataType::QuantisedSymm16)
+    if (!IsQuantizedType(inputTensorInfo.GetDataType()))
     {
         throw InvalidArgumentException(descriptorName + ": Input to dequantize layer must be quantized type.");
     }