From 72c6d9b50b44896a017f2614815c144af229f740 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Principal=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Fri, 9 Nov 2018 17:27:39 +0900 Subject: [PATCH] [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 --- contrib/enco/frontend/tflite/src/Frontend.cpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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; -- 2.7.4