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 | |
| 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
| -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); |