commit bb7b281fb2056eb0124f04014acac48c9c525995
author Colm Donelan <colm.donelan@arm.com> Mon May 09 17:52:23 2022 +0100
committer Colm Donelan <colm.donelan@arm.com> Mon May 09 17:52:23 2022 +0100
tree 7db74289def6fecb327c58dab8cc0c673a6b03ff
parent 915f2a7327098636cc5a292510cd00b715e67e90

IVGCVSW-6127 Fixing seg fault in LayerInputHandle

* Fixing a segmentation fault when SanitizeQuantizationScale is
  called on a LayerInputHandle whose m_OutputSlot is nullptr.

Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: I447c405086ff327a28a72938fbf82844d715a17d

ConversionUtils.cpp

diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp
index 91c1ae4..f441e7d 100644
--- a/ConversionUtils.cpp
+++ b/ConversionUtils.cpp
@@ -62,14 +62,17 @@
 void LayerInputHandle::SanitizeQuantizationScale(LayerInputHandle& weight,
                                                  LayerInputHandle& input)
 {
-    armnn::TensorInfo weightInfo = weight.GetTensorInfo();
-    armnn::TensorInfo inputInfo = input.GetTensorInfo();
-    armnn::TensorInfo biasInfo = GetTensorInfo();
+    if (m_OutputSlot)
+    {
+        armnn::TensorInfo weightInfo = weight.GetTensorInfo();
+        armnn::TensorInfo inputInfo = input.GetTensorInfo();
+        armnn::TensorInfo biasInfo = GetTensorInfo();
 
-    SanitizeBiasQuantizationScale(biasInfo, weightInfo, inputInfo);
+        SanitizeBiasQuantizationScale(biasInfo, weightInfo, inputInfo);
 
-    m_TensorInfo = biasInfo;
-    m_OutputSlot->SetTensorInfo(biasInfo);
+        m_TensorInfo = biasInfo;
+        m_OutputSlot->SetTensorInfo(biasInfo);
+    }
 }
 
 ConstTensorPin::ConstTensorPin(bool optional)