[coco] Support Div operation (#2770)

This commit extends coco IR including Div operation

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>

diff --git a/contrib/coco/core/include/coco/IR/Op.lst b/contrib/coco/core/include/coco/IR/Op.lst
index 4ec8c2e..2bdb7c8 100644 (file)
--- a/contrib/coco/core/include/coco/IR/Op.lst
+++ b/contrib/coco/core/include/coco/IR/Op.lst
@@ -14,4 +14,5 @@ OP(ReLU6)
 OP(Add)
 OP(Sub)
 OP(Mul)
+OP(Div)
 OP(ConcatF)

diff --git a/contrib/coco/core/include/coco/IR/Ops.h b/contrib/coco/core/include/coco/IR/Ops.h
index c79a95e..40cc489 100644 (file)
--- a/contrib/coco/core/include/coco/IR/Ops.h
+++ b/contrib/coco/core/include/coco/IR/Ops.h
@@ -336,6 +336,25 @@ public:
 };
 
 /**
+ * @brief Element-wise division
+ *
+ * Div(L, R) is valid only when L and R have identical kind/shape/dtype
+ */
+class Div final : public BinaryOp
+{
+public:
+  explicit Div() = default;
+
+public:
+  Div(const Div &) = delete;
+  Div(Div &&) = delete;
+
+public:
+  Div *asDiv(void) override { return this; }
+  const Div *asDiv(void) const override { return this; }
+};
+
+/**
  * @brief Concatenate two feature maps
  *
  * ConcatF(L, R) requires

diff --git a/contrib/coco/core/src/IR/OpManager.test.cpp b/contrib/coco/core/src/IR/OpManager.test.cpp
index 62aeb53..0af6ac7 100644 (file)
--- a/contrib/coco/core/src/IR/OpManager.test.cpp
+++ b/contrib/coco/core/src/IR/OpManager.test.cpp
@@ -68,6 +68,13 @@ TEST_F(OpManagerTest, Sub)
   ASSERT_NE(obj, nullptr);
 }
 
+TEST_F(OpManagerTest, Div)
+{
+  auto obj = mgr.create<coco::Div>();
+
+  ASSERT_NE(obj, nullptr);
+}
+
 TEST_F(OpManagerTest, PadF)
 {
   auto op = mgr.create<coco::PadF>();

diff --git a/contrib/coco/core/src/IR/Ops.test.cpp b/contrib/coco/core/src/IR/Ops.test.cpp
index 965708b..ae979b2 100644 (file)
--- a/contrib/coco/core/src/IR/Ops.test.cpp
+++ b/contrib/coco/core/src/IR/Ops.test.cpp
@@ -78,6 +78,17 @@ TEST(MulTest, constructor)
 }
 
 /**
+ * Section: Div Op
+ */
+TEST(DivTest, constructor)
+{
+  auto op = make_unique<coco::Div>();
+
+  ASSERT_EQ(op->left(), nullptr);
+  ASSERT_EQ(op->right(), nullptr);
+}
+
+/**
  * Section: Op Helpers
  */
 namespace