aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2022-06-21 11:39:41 +0100
committerSadik Armagan <sadik.armagan@arm.com>2022-06-21 13:06:16 +0000
commitb9570c1a5b938aea20964e3b6a8164e9bf096b4c (patch)
treefa5e8784415dcd91c630c25ce020fe87b0381bba
parenta0f8b15d4ddb5075f380003ff31b271d389d3b66 (diff)
downloadarmnn-b9570c1a5b938aea20964e3b6a8164e9bf096b4c.tar.gz
IVGCVSW-6954 'Fixed test failures on Conv2d"
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: Ifd5ad4d4ca553e3f58ea246a0ac3c6e6b8b31215
-rw-r--r--shim/sl/canonical/ConversionUtils.cpp5
1 files 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
{