IVGCVSW-5418 'ExecuteNetwork test for MobileBERT'
* Refactored the code for checking constant inputs.
* Added a unit test for ADD operator with constant input.
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ie7207e5a1ce77ea305552859de32a66e07c68a6f
diff --git a/delegate/src/Gather.hpp b/delegate/src/Gather.hpp
index 9ed0fe1..634373a 100644
--- a/delegate/src/Gather.hpp
+++ b/delegate/src/Gather.hpp
@@ -87,9 +87,17 @@
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;