diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2022-06-21 11:39:41 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2022-06-21 13:06:16 +0000 |
commit | b9570c1a5b938aea20964e3b6a8164e9bf096b4c (patch) | |
tree | fa5e8784415dcd91c630c25ce020fe87b0381bba /shim | |
parent | a0f8b15d4ddb5075f380003ff31b271d389d3b66 (diff) | |
download | armnn-b9570c1a5b938aea20964e3b6a8164e9bf096b4c.tar.gz |
IVGCVSW-6954 'Fixed test failures on Conv2d"
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ifd5ad4d4ca553e3f58ea246a0ac3c6e6b8b31215
Diffstat (limited to 'shim')
-rw-r--r-- | shim/sl/canonical/ConversionUtils.cpp | 5 |
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 { |