[coco] Record output name (#783)

This commit revises 'Output' class to record its name.

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

diff --git a/contrib/coco/core/include/coco/IR/Output.h b/contrib/coco/core/include/coco/IR/Output.h
index 2748dbd..8a48470 100644 (file)
--- a/contrib/coco/core/include/coco/IR/Output.h
+++ b/contrib/coco/core/include/coco/IR/Output.h
@@ -5,6 +5,8 @@
 
 #include <nncc/core/ADT/tensor/Shape.h>
 
+#include <string>
+
 namespace coco
 {
 
@@ -25,6 +27,13 @@ public:
 
 private:
   Bag *_bag;
+
+public:
+  const std::string &name(void) const { return _name; }
+  void name(const std::string &s) { _name = s; }
+
+private:
+  std::string _name;
 };
 
 } // namespace coco

diff --git a/contrib/coco/core/src/IR/Output.test.cpp b/contrib/coco/core/src/IR/Output.test.cpp
index fe804ff..2f00923 100644 (file)
--- a/contrib/coco/core/src/IR/Output.test.cpp
+++ b/contrib/coco/core/src/IR/Output.test.cpp
@@ -20,3 +20,12 @@ TEST(IR_OUTPUT, bag_update)
   output.bag(&bag);
   ASSERT_EQ(output.bag(), &bag);
 }
+
+TEST(IR_OUTPUT, name_update)
+{
+  const nncc::core::ADT::tensor::Shape shape{1, 3, 3, 1};
+  coco::Output output{shape};
+
+  output.name("softmax");
+  ASSERT_EQ(output.name(), "softmax");
+}