[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 (file)
--- 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;