From 915f2a7327098636cc5a292510cd00b715e67e90 Mon Sep 17 00:00:00 2001
From: Cathal Corbett <cathal.corbett@arm.com>
Date: Fri, 15 Apr 2022 14:12:08 +0100
Subject: IVGCVSW-6127 ConstTensorsAsInput: DepthwiseConvolution2d

!armnn:7417

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ic37d6b8677e040c60a90358dd0a4a8eb33fb6ea0
---
 ConversionUtils.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'ConversionUtils.cpp')

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)
 {}
-- 
cgit v1.2.1