Add layer examples

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,9 @@
+set(ARM_DIR "/3rdparty/ComputeLibrary")
+
+include_directories(${ARM_DIR}/include)
+link_directories(${ARM_DIR}/build)
+
+add_executable(ConvolutionLayer ConvolutionLayer.cpp)
+add_executable(ElementwiseLayer ElementwiseLayer.cpp)
+
+target_link_libraries(ConvolutionLayer arm_compute)

app/layer_example/ConvolutionLayer.cpp

@@ -0,0 +1,40 @@
+#include "../ComputeLibrary/arm_compute/runtime/NEON/NEFunctions.h"
+#include "../ComputeLibrary/utils/Utils.h"
+
+#include <iostream>
+using namespace arm_compute;
+using namespace utils;
+
+int main(){
+    Tensor input;
+    Tensor weight;
+    Tensor bias;
+    Tensor output;
+
+    const unsigned int N = 1;
+    const unsigned int Hin = 3;
+    const unsigned int Win = 3;
+    const unsigned int Cin = 1;
+
+    const unsigned int Hf = 3;
+    const unsigned int Wf = 3;
+
+    const unsigned int Hout = Hin - Hf + 1;
+    const unsigned int Wout = Win - Wf + 1;
+    const unsigned int Cout = 1;
+
+    input.allocator()->init(TensorInfo(TensorShape(Hin, Win, Cin), 1, DataType::F32));
+    weight.allocator()->init(TensorInfo(TensorShape(Hf, Wf, Cin, Cout), 1, DataType::F32));
+    output.allocator()->init(TensorInfo(TensorShape(Hout, Wout, Cout), 1, DataType::F32));
+
+    input.allocator()->allocate();
+    weight.allocator()->allocate();
+    output.allocator()->allocate();
+
+    NEConvolutionLayer conv;
+    conv.configure(&input, &weight, nullptr, &output, PadStrideInfo(1, 1, 0, 0));
+
+    conv.run();
+
+    output.print(std::cout);
+}

app/layer_example/ElementwiseLayer.cpp

@@ -0,0 +1,31 @@
+#include "../ComputeLibrary/arm_compute/runtime/NEON/functions/NEElementwiseOperations.h"
+#include "../ComputeLibrary/utils/Utils.h"
+
+#include <iostream>
+using namespace arm_compute;
+using namespace utils;
+
+int main() {
+    const int input_width = 5;
+    const int input_height = 5;
+
+    Tensor input1, input2, output;
+
+    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);
+
+    NEElementwiseSquaredDiff elementwise;
+    elementwise.configure(&input1, &input2, &output);
+
+    elementwise.run();
+
+    output.print(std::cout);
+}