From 6726b0f4a57d8cfeeedce085d1a44bdcafbb07b3 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EC=9C=A4=ED=98=84=EC=8B=9D/On-Device=20Lab=28SR=29/Princip?=
 =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?=
 <hyunsik.yoon@samsung.com>
Date: Fri, 2 Aug 2019 08:48:19 +0900
Subject: [PATCH] [moco-tf] Enable moco to print custom op log (#6114)

By adding `COpNodeSummaryBuilder`, moco can print custom op info.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
---
 compiler/moco-tf/CMakeLists.txt           | 2 ++
 compiler/moco-tf/requires.cmake           | 1 +
 compiler/moco-tf/src/TFFormattedGraph.cpp | 6 ++++++
 3 files changed, 9 insertions(+)

diff --git a/compiler/moco-tf/CMakeLists.txt b/compiler/moco-tf/CMakeLists.txt
index 89fcf88..254fb16 100644
--- a/compiler/moco-tf/CMakeLists.txt
+++ b/compiler/moco-tf/CMakeLists.txt
@@ -29,6 +29,7 @@ target_link_libraries(moco_tf_frontend PRIVATE pepper_str)
 target_link_libraries(moco_tf_frontend PRIVATE pepper_strcast)
 target_link_libraries(moco_tf_frontend PRIVATE locomotiv)
 target_link_libraries(moco_tf_frontend PRIVATE plier_tf)
+target_link_libraries(moco_tf_frontend PRIVATE locoex_customop)
 
 if(NOT ENABLE_TEST)
   return()
@@ -44,4 +45,5 @@ target_link_libraries(moco_tf_frontend_test moco_log)
 target_link_libraries(moco_tf_frontend_test moco_tf_frontend)
 target_link_libraries(moco_tf_frontend_test stdex)
 target_link_libraries(moco_tf_frontend_test plier_tf)
+target_link_libraries(moco_tf_frontend_test locoex_customop)
 add_test(moco_tf_frontend_test moco_tf_frontend_test)
diff --git a/compiler/moco-tf/requires.cmake b/compiler/moco-tf/requires.cmake
index 961b5b2..be91c8c 100644
--- a/compiler/moco-tf/requires.cmake
+++ b/compiler/moco-tf/requires.cmake
@@ -6,3 +6,4 @@ require("moco-log")
 require("pepper-strcast")
 require("locomotiv")
 require("plier-tf")
+require("locoex-customop")
diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp
index 399c2aa..bfec20a 100644
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -24,6 +24,7 @@
 #include "LogHelper.h"
 
 #include <pepper/str.h>
+#include <locoex/Service/COpFormattedGraph.h>
 
 #include <sstream>
 
@@ -252,6 +253,11 @@ bool MocoNodeSummaryBuilder::build(const loco::Node *node, locop::NodeSummary &s
     return true;
   }
 
+  if (locoex::COpNodeSummaryBuilder(_tbl).build(node, s))
+  {
+    return true;
+  }
+
   return false;
 }
 
-- 
2.7.4