[enco/tfl/frontend] TensorContext returns tensor type (#2533)

This commit enables TensorContext to return tensor type.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>

diff --git a/contrib/enco/frontend/tflite/src/Context.cpp b/contrib/enco/frontend/tflite/src/Context.cpp
index cfdbc06..ef030dc 100644 (file)
--- a/contrib/enco/frontend/tflite/src/Context.cpp
+++ b/contrib/enco/frontend/tflite/src/Context.cpp
@@ -39,9 +39,11 @@ void TensorContext::prepare(const tflite::SubGraph *graph)
     auto const tensor_info = graph->tensors()->Get(tensor_id);
     auto const tensor_name = tensor_info->name()->str();
     auto const tensor_shape = as_tensor_shape(tensor_info->shape());
+    auto const tensor_type = tensor_info->type();
 
     _name_ctx[tensor_id] = tensor_name;
     _shape_ctx[tensor_id] = tensor_shape;
+    _type_ctx[tensor_id] = tensor_type;
   }
 }
 

diff --git a/contrib/enco/frontend/tflite/src/Context.h b/contrib/enco/frontend/tflite/src/Context.h
index d126a02..f72385f 100644 (file)
--- a/contrib/enco/frontend/tflite/src/Context.h
+++ b/contrib/enco/frontend/tflite/src/Context.h
@@ -33,7 +33,7 @@ namespace tflimport
 {
 
 /**
- * @brief Extracts and holds operand(tensor) information such as name and shape
+ * @brief Extracts and holds operand(tensor) information such as name, shape, and type
  */
 class TensorContext
 {
@@ -42,10 +42,12 @@ public:
 
   const std::string &name(uint32_t tensor_id) { return _name_ctx[tensor_id]; }
   const tensor::Shape &shape(uint32_t tensor_id) { return _shape_ctx[tensor_id]; }
+  const tflite::TensorType &type(uint32_t tensor_id) { return _type_ctx[tensor_id]; }
 
 private:
   std::map<uint32_t, std::string> _name_ctx;
   std::map<uint32_t, tensor::Shape> _shape_ctx;
+  std::map<uint32_t, tflite::TensorType> _type_ctx;
 };
 
 /**