From: 박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Date: Fri, 9 Nov 2018 08:27:39 +0000 (+0900)
Subject: [enco/frontend] Set module inputs and outputs (#2193)
X-Git-Tag: nncc_backup~1382
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=72c6d9b50b44896a017f2614815c144af229f740;p=platform%2Fcore%2Fml%2Fnnfw.git

[enco/frontend] Set module inputs and outputs (#2193)

This will set coco Module inputs and outpus from TF lite graph

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
---

diff --git a/contrib/enco/frontend/tflite/src/Frontend.cpp b/contrib/enco/frontend/tflite/src/Frontend.cpp
index d436b6f..9bab42a 100644
--- a/contrib/enco/frontend/tflite/src/Frontend.cpp
+++ b/contrib/enco/frontend/tflite/src/Frontend.cpp
@@ -269,6 +269,18 @@ enco::Bundle Frontend::load(void) const
   auto m = coco::Module::create();
   auto d = coco::Data::create();
 
+  tflimport::TensorContext tensor_context;
+  tflimport::TensorBags tensor_bags;
+
+  tensor_context.prepare(graph);
+  tensor_bags.prepare(graph, m);
+
+  auto inputs = tflimport::as_index_vector(graph->inputs());
+  auto outputs = tflimport::as_index_vector(graph->outputs());
+
+  tflimport::set_module_inputs(m.get(), tensor_context, tensor_bags, inputs);
+  tflimport::set_module_outputs(m.get(), tensor_context, tensor_bags, outputs);
+
   // Create "Bundle"
   enco::Bundle bundle;