1 files changed, 9 insertions, 1 deletions
diff --git a/delegate/src/Gather.hpp b/delegate/src/Gather.hpp
index 9ed0fe15c1..634373a341 100644
--- a/delegate/src/Gather.hpp
+++ b/delegate/src/Gather.hpp
@@ -87,9 +87,17 @@ TfLiteStatus VisitGatherOperator(DelegateData& delegateData,
 
     armnn::IConnectableLayer* layer = delegateData.m_Network->AddGatherLayer(gatherDescriptor);
     ARMNN_ASSERT(layer != nullptr);
-
     layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
 
+    auto inputsTensorsProcess = ProcessInputs(layer,
+                                              delegateData,
+                                              tfLiteContext,
+                                              tfLiteNode);
+    if (inputsTensorsProcess == kTfLiteError)
+    {
+        return inputsTensorsProcess;
+    }
+
     Connect(layer, tfLiteNode, delegateData);
 
     return kTfLiteOk;