[Pure ACL NN Runtime] Implement Compilation (#522)

This commit partially implements ANeuralNetworksCompilation_create with
partially defined ANeuralNetworksCompilation.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.cc b/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.cc
index 98b0722..d046f8b 100644 (file)
--- a/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.cc
+++ b/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.cc
@@ -1,10 +1,17 @@
 #include <nnapi.h>
 
 #include "compilation.h"
+#include "model.h"
 
 ResultCode
 ANeuralNetworksCompilation_create(ANeuralNetworksModel* model, ANeuralNetworksCompilation** compilation)
 {
+  std::shared_ptr<const internal::tflite::Model> internal;
+
+  model->release(internal);
+
+  *compilation = new ANeuralNetworksCompilation(internal);
+
   return ANEURALNETWORKS_NO_ERROR;
 }
 

diff --git a/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.h b/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.h
index 7a28742..f14f631 100644 (file)
--- a/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.h
+++ b/tools/nnapi_bindings/bindings/pure_arm_compute/src/compilation.h
@@ -1,8 +1,22 @@
 #ifndef __COMPILATION_H__
 #define __COMPILATION_H__
 
+#include "internal/Model.h"
+
 struct ANeuralNetworksCompilation
 {
+public:
+  ANeuralNetworksCompilation(const std::shared_ptr<const internal::tflite::Model> &model)
+    : _model{model}
+  {
+    // DO NOTHING
+  }
+
+public:
+  const internal::tflite::Model &model(void) const { return *_model; }
+
+private:
+  std::shared_ptr<const internal::tflite::Model> _model;
 };
 
 #endif