diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2022-01-18 14:31:05 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2022-01-18 17:41:06 +0000 |
commit | adeebaa73205bd981ea7e8c8f135f01355cba841 (patch) | |
tree | 82ddedeed70182310b7fc69330c486f335dc5769 /delegate | |
parent | 529195f3ae752192d5f205345235bf537f472cbc (diff) | |
download | armnn-adeebaa73205bd981ea7e8c8f135f01355cba841.tar.gz |
IVGCVSW-6717 'Inference Passes on Tflite-Parser but fails when using Arm NN Delegate'
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I40ab6aff4bc076eafcef08c1c20aacdbd52da853
Diffstat (limited to 'delegate')
-rw-r--r-- | delegate/src/DelegateUtils.hpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/delegate/src/DelegateUtils.hpp b/delegate/src/DelegateUtils.hpp index e0de809ab3..678a3db002 100644 --- a/delegate/src/DelegateUtils.hpp +++ b/delegate/src/DelegateUtils.hpp @@ -222,6 +222,7 @@ armnn::IConnectableLayer* BroadcastTensor(const armnn::TensorInfo& inputInfo0, reshapedDimensions.data() }); armnn::ReshapeDescriptor reshapeDescriptor; + reshapeDescriptor.m_TargetShape = reshapedInfo.GetShape(); bool isSupported = false; FORWARD_LAYER_SUPPORT_FUNC(__func__, tfLiteContext, @@ -238,8 +239,6 @@ armnn::IConnectableLayer* BroadcastTensor(const armnn::TensorInfo& inputInfo0, ARMNN_ASSERT(delegateData.m_Network != nullptr); // Add Reshape layer - reshapeDescriptor.m_TargetShape = reshapedInfo.GetShape(); - armnn::IConnectableLayer* reshapeLayer = delegateData.m_Network->AddReshapeLayer(reshapeDescriptor); ARMNN_ASSERT(reshapeLayer != nullptr); reshapeLayer->GetOutputSlot(0).SetTensorInfo(reshapedInfo); |