[loco] Introducing BiasDecode (#7158)

This introduce loco::BiasDecode, which translate Bias domain to Tensor domain.

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

diff --git a/compiler/loco/include/loco/IR/CanonicalNodes.lst b/compiler/loco/include/loco/IR/CanonicalNodes.lst
index 2acc259..2bd2cf0 100644 (file)
--- a/compiler/loco/include/loco/IR/CanonicalNodes.lst
+++ b/compiler/loco/include/loco/IR/CanonicalNodes.lst
@@ -8,6 +8,7 @@
 
 // CANONICAL_NODE(OPCODE, CLASS)
 CANONICAL_NODE(AvgPool2D, AvgPool2D)
+CANONICAL_NODE(BiasDecode, BiasDecode)
 CANONICAL_NODE(BiasEncode, BiasEncode)
 CANONICAL_NODE(ConstGen, ConstGen)
 CANONICAL_NODE(Conv2D, Conv2D)

diff --git a/compiler/loco/include/loco/IR/Nodes.h b/compiler/loco/include/loco/IR/Nodes.h
index 11bdc7a..813bfe2 100644 (file)
--- a/compiler/loco/include/loco/IR/Nodes.h
+++ b/compiler/loco/include/loco/IR/Nodes.h
@@ -593,6 +593,19 @@ private:
 using TensorSoftmax = Softmax<Domain::Tensor>;
 
 /**
+ * @brief Create a "Tensor" from a "Bias"
+ */
+class BiasDecode final : public CanonicalNodeDef<CanonicalOpcode::BiasDecode, FixedArity<1>::Mixin>
+{
+public:
+  BiasDecode() = default;
+
+public:
+  Node *input(void) const { return at(0)->node(); }
+  void input(Node *node) { at(0)->node(node); }
+};
+
+/**
  * @brief Create a "Bias" from a "Tensor"
  *
  * BiasEncode currently requires a rank-1 tensor as its input.