From b9570c1a5b938aea20964e3b6a8164e9bf096b4c Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Tue, 21 Jun 2022 11:39:41 +0100 Subject: IVGCVSW-6954 'Fixed test failures on Conv2d" Signed-off-by: Sadik Armagan Change-Id: Ifd5ad4d4ca553e3f58ea246a0ac3c6e6b8b31215 --- shim/sl/canonical/ConversionUtils.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/shim/sl/canonical/ConversionUtils.cpp b/shim/sl/canonical/ConversionUtils.cpp index fbc5e280d3..020410d30e 100644 --- a/shim/sl/canonical/ConversionUtils.cpp +++ b/shim/sl/canonical/ConversionUtils.cpp @@ -286,9 +286,10 @@ LayerInputHandle ConvertToLayerInputHandle(const Operation& operation, armnn::IConnectableLayer* constantLayer = data.m_Network->AddConstantLayer(tensorPin.GetConstTensor()); armnn::IOutputSlot& outputSlot = constantLayer->GetOutputSlot(0); - outputSlot.SetTensorInfo(tensorPin.GetConstTensor().GetInfo()); + armnn::TensorInfo constantTensorInfo = tensorPin.GetConstTensor().GetInfo(); + outputSlot.SetTensorInfo(constantTensorInfo); - return LayerInputHandle(true, &outputSlot, operandTensorInfo); + return LayerInputHandle(true, &outputSlot, constantTensorInfo); } else { -- cgit v1.2.1