Add a MatMul Layer example

app/CMakeLists.txt

@@ -1 +1,2 @@
 add_subdirectory(example)
+add_subdirectory(layer_example)

app/example/CMakeLists.txt

@@ -1 +1 @@
-add_executable(example main.cpp)
+add_executable(example main.cpp)

app/layer_example/CMakeLists.txt

@@ -0,0 +1,11 @@
+set(ARM_DIR "${CMAKE_SOURCE_DIR}/3rdparty/ComputeLibrary")
+
+add_executable(MatMul MatMulLayer.cpp)
+
+include_directories(${ARM_DIR})
+include_directories(${ARM_DIR}/include)
+target_link_directories(MatMul PUBLIC ${ARM_DIR}/build)
+
+target_link_libraries(MatMul arm_compute)
+
+add_dependencies(MatMul build_compute_library)

app/layer_example/MatMulLayer.cpp

@@ -0,0 +1,32 @@
+#include "arm_compute/runtime/NEON/NEFunctions.h"
+#include "utils/Utils.h"
+
+#include <iostream>
+using namespace arm_compute;
+using namespace utils;
+
+int main() {
+    Tensor input1;
+    Tensor input2;
+    Tensor output;
+
+    const int input_width  = 5;
+    const int input_height = 5;
+
+    input1.allocator()->init(TensorInfo(TensorShape(input_width, input_height, 1), 1, DataType::F32));
+    input2.allocator()->init(TensorInfo(TensorShape(input_width, input_height, 1), 1, DataType::F32));
+    output.allocator()->init(TensorInfo(TensorShape(input_width, input_height, 1), 1, DataType::F32));
+
+    input1.allocator()->allocate();
+    input2.allocator()->allocate();
+    output.allocator()->allocate();
+
+    fill_random_tensor(input1, 0.f, 1.f);
+    fill_random_tensor(input2, 0.f, 1.f);
+
+    NEMatMul m; 
+    m.configure(&input1, &input2, &output, MatMulInfo(), CpuMatMulSettings(), ActivationLayerInfo());
+    m.run();
+
+    output.print(std::cout);
+}