aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/WorkloadData.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
-rw-r--r--src/backends/backendsCommon/WorkloadData.cpp5
1 files changed, 1 insertions, 4 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 83e823c9a5..6d5c6e8343 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -2483,10 +2483,7 @@ void DequantizeQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
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.");
}