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/ElementwiseBinary.hpp b/delegate/src/ElementwiseBinary.hpp
index 49a5dfb..58d7aca 100644
--- a/delegate/src/ElementwiseBinary.hpp
+++ b/delegate/src/ElementwiseBinary.hpp
@@ -273,32 +273,13 @@
armnn::IOutputSlot& outputSlot = elementwiseBinaryLayer->GetOutputSlot(0);
outputSlot.SetTensorInfo(outputTensorInfo);
- if(tflite::IsConstantTensor(&tfLiteInputTensor0))
+ auto inputsTensorsProcess = ProcessInputs(elementwiseBinaryLayer,
+ delegateData,
+ tfLiteContext,
+ tfLiteNode);
+ if (inputsTensorsProcess == kTfLiteError)
{
- auto status = ConnectConstant(elementwiseBinaryLayer,
- inputTensorInfo0,
- tfLiteContext,
- tfLiteInputTensor0,
- delegateData,
- tfLiteNode->inputs->data[0]);
- if (status == kTfLiteError)
- {
- return status;
- }
- }
-
- if(tflite::IsConstantTensor(&tfLiteInputTensor1))
- {
- auto status = ConnectConstant(elementwiseBinaryLayer,
- inputTensorInfo1,
- tfLiteContext,
- tfLiteInputTensor1,
- delegateData,
- tfLiteNode->inputs->data[1]);
- if (status == kTfLiteError)
- {
- return status;
- }
+ return inputsTensorsProcess;
}
auto reshapeLayer = BroadcastTensor(inputTensorInfo0,