diff options
Diffstat (limited to 'ConversionUtils.hpp')
| -rw-r--r-- | ConversionUtils.hpp | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index a284a50a..6f1f100d 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -2220,6 +2220,13 @@ bool ConvertDequantize(const Operation& operation, const Model& model, Conversio return Fail("%s: Operation has invalid input", __func__); } + const armnn::TensorInfo& inputInfo = input.GetTensorInfo(); + const armnn::Optional<unsigned int>& quantizationDim = inputInfo.GetQuantizationDim(); + if (quantizationDim.has_value() && quantizationDim.value() != 0) + { + return Fail("%s: Operation has quantization dimension different than 0", __func__); + } + const Operand* const outputOperand = GetOutputOperand<HalPolicy>(operation, 0, model); if (!outputOperand) { @@ -2237,8 +2244,8 @@ bool ConvertDequantize(const Operation& operation, const Model& model, Conversio IsDequantizeSupported, data.m_Backends, isSupported, - input.GetTensorInfo(), - GetTensorInfoForOperand(*outputOperand)); + inputInfo, + outputInfo); if (!isSupported) { return false; |