commit 90a119b9c992de662d3719b04f03735b96259f65
author Sadik Armagan <sadik.armagan@arm.com> Fri Aug 05 16:12:49 2022 +0100
committer Nikhil Raj <nikhil.raj@arm.com> Mon Aug 29 10:11:09 2022 +0100
tree 1b4ea68c8f66e726373814682e747cea8f50be51
parent c814e8031dfe9565e94cf89ef193b6af944c73b1

IVGCVSW-7154 'Constant Tensors As Inputs'

* Fixed the issues about converting Conv2D and DepthwiseConv2d input issues
* Read 1D input tensors that do not have shape specified

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I12f3d1c57a2afedac42c6e7b31e4b1fc689abeca

delegate/src/Gather.hpp

diff --git a/delegate/src/Gather.hpp b/delegate/src/Gather.hpp
index 11f3b61..616de7e 100644
--- a/delegate/src/Gather.hpp
+++ b/delegate/src/Gather.hpp
@@ -47,7 +47,7 @@
 
     const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor);
     const armnn::TensorInfo& indicesTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteIndicesTensor);
-    const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor);
+    const armnn::TensorInfo& outputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputTensor, true);
     armnn::GatherDescriptor gatherDescriptor;
     gatherDescriptor.m_Axis = axis;