From adeebaa73205bd981ea7e8c8f135f01355cba841 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Tue, 18 Jan 2022 14:31:05 +0000 Subject: IVGCVSW-6717 'Inference Passes on Tflite-Parser but fails when using Arm NN Delegate' Signed-off-by: Sadik Armagan Change-Id: I40ab6aff4bc076eafcef08c1c20aacdbd52da853 --- delegate/src/DelegateUtils.hpp | 3 +-- 1 file changed, 1 insertion(+), 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); -- cgit v1.2.1