diff options
Diffstat (limited to 'ConversionUtils.cpp')
-rw-r--r-- | ConversionUtils.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp index 7eb7343e..91c1ae4e 100644 --- a/ConversionUtils.cpp +++ b/ConversionUtils.cpp @@ -59,6 +59,19 @@ const armnn::TensorInfo& LayerInputHandle::GetTensorInfo() const return m_TensorInfo; } +void LayerInputHandle::SanitizeQuantizationScale(LayerInputHandle& weight, + LayerInputHandle& input) +{ + armnn::TensorInfo weightInfo = weight.GetTensorInfo(); + armnn::TensorInfo inputInfo = input.GetTensorInfo(); + armnn::TensorInfo biasInfo = GetTensorInfo(); + + SanitizeBiasQuantizationScale(biasInfo, weightInfo, inputInfo); + + m_TensorInfo = biasInfo; + m_OutputSlot->SetTensorInfo(biasInfo); +} + ConstTensorPin::ConstTensorPin(bool optional) : m_Optional(optional) {} |