commit 15f7faef88357679064e0e9d3bd91dd18c7625d6
author Sadik Armagan <sadik.armagan@arm.com> Wed Nov 18 09:37:03 2020 +0000
committer Sadik Armagan <sadik.armagan@arm.com> Wed Nov 18 09:37:03 2020 +0000
tree d8dcfc49dc67614feb5edf0dc1ff49b7ca2581d4
parent 5d03e31aaf4d82e9f9cdc03c41d2328bbb2a0dee

IVGCVSW-5377 'Add ArmNN TfLite delegate to ExecuteNetwork'

* Assign correct input values for the model
* Call the right Validate function for Mul and Sub operators
* Return the correct data type for kTfLiteInt8

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

delegate/src/ElementwiseBinary.hpp

diff --git a/delegate/src/ElementwiseBinary.hpp b/delegate/src/ElementwiseBinary.hpp
index e527005..49a5dfb 100644
--- a/delegate/src/ElementwiseBinary.hpp
+++ b/delegate/src/ElementwiseBinary.hpp
@@ -228,13 +228,13 @@
                                                inputTensorInfo1,
                                                outputTensorInfo);
             case kTfLiteBuiltinMul:
-                return ValidateDivOperator(delegateData,
+                return ValidateMulOperator(delegateData,
                                            tfLiteContext,
                                            inputTensorInfo0,
                                            inputTensorInfo1,
                                            outputTensorInfo);
             case kTfLiteBuiltinSub:
-                return ValidateDivOperator(delegateData,
+                return ValidateSubOperator(delegateData,
                                            tfLiteContext,
                                            inputTensorInfo0,
                                            inputTensorInfo1,