From bb7b281fb2056eb0124f04014acac48c9c525995 Mon Sep 17 00:00:00 2001 From: Colm Donelan Date: Mon, 9 May 2022 17:52:23 +0100 Subject: 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 Change-Id: I447c405086ff327a28a72938fbf82844d715a17d --- ConversionUtils.cpp | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp index 91c1ae4e..f441e7dc 100644 --- a/ConversionUtils.cpp +++ b/ConversionUtils.cpp @@ -62,14 +62,17 @@ const armnn::TensorInfo& LayerInputHandle::GetTensorInfo() const 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) -- cgit v1.2.1