diff options
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp index c145c4b39f..83e823c9a5 100644 --- a/src/backends/backendsCommon/WorkloadData.cpp +++ b/src/backends/backendsCommon/WorkloadData.cpp @@ -2484,6 +2484,8 @@ void DequantizeQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; if (inputTensorInfo.GetDataType() != DataType::QuantisedAsymm8 && + inputTensorInfo.GetDataType() != DataType::QuantisedSymm8 && + inputTensorInfo.GetDataType() != DataType::QuantizedSymm8PerAxis && inputTensorInfo.GetDataType() != DataType::QuantisedSymm16) { throw InvalidArgumentException(descriptorName + ": Input to dequantize layer must be quantized type."); |