diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 5 |
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."); } |