commit 49ed0df12338b1e99674edeee4200acf8c05750e
author Ryan OShea <ryan.oshea3@arm.com> Wed Sep 21 16:09:41 2022 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Thu Sep 22 10:55:33 2022 +0000
tree 85f0806dde1d8f24c74a986d732e91904da5899a
parent 9636a9b109fcbc811ec876ba9ca6512b7fbe2ba0

IVGCVSW-6498 Add Support for Batch MatMul to TfLite Delegate

 * Creates delegate/src/BatchMatMul.hpp
 * Add VisitBatchMatMul function
 * Add BatchMatMul to switch in armnn_delegate
 * Creates delegate/src/test/BatchMatMulTest.cpp
 * Creates delegate/src/test/BatchMatMulTestHelper.hpp
 * Add Int8 and Fp32 unit tests on ref backend
 * Add BatchMatMul to delegate supported ops

Signed-off-by: Ryan OShea <ryan.oshea3@arm.com>
Change-Id: I50e61314cf063f986c8a0f7d508847a96953735e

delegate/src/armnn_delegate.cpp

diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index c041dd1..21c66fe 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -9,6 +9,7 @@
 
 #include "Activation.hpp"
 #include "ArgMinMax.hpp"
+#include "BatchMatMul.hpp"
 #include "BatchSpace.hpp"
 #include "Comparison.hpp"
 #include "Convolution.hpp"
@@ -566,6 +567,12 @@
                                         tfLiteNode,
                                         nodeIndex,
                                         kTfLiteBuiltinAveragePool2d);
+        case kTfLiteBuiltinBatchMatmul:
+            return VisitBatchMatMulOperator(delegateData,
+                                            tfLiteContext,
+                                            tfLiteNode,
+                                            nodeIndex,
+                                            kTfLiteBuiltinBatchMatmul);
         case kTfLiteBuiltinBatchToSpaceNd:
             return VisitBatchToSpaceNdOperator(delegateData,
                                                tfLiteContext,