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)
 {}