commit 915f2a7327098636cc5a292510cd00b715e67e90
author Cathal Corbett <cathal.corbett@arm.com> Fri Apr 15 14:12:08 2022 +0100
committer Cathal Corbett <cathal.corbett@arm.com> Thu May 05 09:11:47 2022 +0000
tree 20933b85c8e2f46f97c82031b7925c176d4ccf7b
parent 0ae102a0fc9d66b7067cf8d7a0ed1af5ed65ae50

IVGCVSW-6127 ConstTensorsAsInput: DepthwiseConvolution2d

!armnn:7417

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ic37d6b8677e040c60a90358dd0a4a8eb33fb6ea0

ConversionUtils.cpp

diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp
index 7eb7343..91c1ae4 100644
--- a/ConversionUtils.cpp
+++ b/ConversionUtils.cpp
@@ -59,6 +59,19 @@
     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)
 {}