commit 4b7a34dd92eb3f736e05ac6623fd147ecd8636b1
author Samuel Yap <samuel.yap@arm.com> Wed Jul 06 15:36:03 2022 +0100
committer Samuel Yap <samuel.yap@arm.com> Fri Jul 22 16:52:38 2022 +0100
tree c33e5820f89e359c80d8773288e8adb075735039
parent 16929a2b432232f7a34fcbd1f1b0fe1212500206

IVGCVSW-7109: Add Batch MatMul front end support - Reference

  * Descriptors added for BatchMatMul
  * Layer definition added
  * Input validation added (will likely change when opt. param support comes in)
  * Ref workload implementation for BatchMatMul added (will also change with opt. param support)
  * Ref layer tests made for BatchMatMul
  * CMake and other build files updated

Signed-off-by: Samuel Yap <samuel.yap@arm.com>
Change-Id: Ic885301da543ee0fbe7922b85e7f9658c4efc617

CMakeLists.txt

diff --git a/CMakeLists.txt b/CMakeLists.txt
index f0eb81c..1d8ebe2 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -187,6 +187,8 @@
     src/armnn/layers/AdditionLayer.cpp
     src/armnn/layers/ArgMinMaxLayer.hpp
     src/armnn/layers/ArgMinMaxLayer.cpp
+    src/armnn/layers/BatchMatMulLayer.hpp
+    src/armnn/layers/BatchMatMulLayer.cpp
     src/armnn/layers/BatchNormalizationLayer.hpp
     src/armnn/layers/BatchNormalizationLayer.cpp
     src/armnn/layers/BatchToSpaceNdLayer.hpp