[locop] Add TensorReduce visit function (#7597)

* [locop] Add TensorReduce visit function

This commit will add `TensorReduce` visit function

Signed-off-by: Seok NamKoong <sk.namkoong@samsung.com>
* apply comments

* fix mistake

diff --git a/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp b/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp
index 2c135c8..b962f49 100644 (file)
--- a/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp
+++ b/compiler/locop/src/CanonicalNodeSummaryBuilder.cpp
@@ -217,6 +217,19 @@ public:
     return res;
   }
 
+  NodeDesc visit(const loco::TensorReduce *node) final
+  {
+    NodeDesc res{opname(node)};
+
+    // TODO Print TensorAxisSet
+    res.arg("input", nodename(node->input()));
+    res.arg("func", pp::fmt((int32_t)node->func()));
+
+    res.state(NodeDesc::State::PartiallyKnown);
+
+    return res;
+  }
+
   NodeDesc visit(const loco::Reshape<loco::ReshapeType::Fixed> *node) final
   {
     NodeDesc res{opname(node)};