[moco-tf] Add TFFormattedGraph for TFTanh (#6850)
author남궁석/On-Device Lab(SR)/Engineer/삼성전자 <sk.namkoong@samsung.com>
Fri, 23 Aug 2019 01:01:40 +0000 (10:01 +0900)
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Fri, 23 Aug 2019 01:01:40 +0000 (10:01 +0900)
This commit will add `TFFormattedGraph` for `TFTanh`

Signed-off-by: Seok NamKoong <sk.namkoong@samsung.com>
compiler/moco-tf/src/TFFormattedGraph.cpp

index a99f6df..fdb4f33 100644 (file)
@@ -91,6 +91,7 @@ private:
   IMPLEMENT(TFSoftmax);
   IMPLEMENT(TFSqueeze);
   IMPLEMENT(TFStopGradient);
+  IMPLEMENT(TFTanh);
 #undef IMPLEMENT
 };
 
@@ -291,6 +292,14 @@ bool TFNodeSummaryBuilder::summary(const TFStopGradient *node, locop::NodeSummar
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFTanh *node, locop::NodeSummary &s) const
+{
+  s.opname("TF.Tanh");
+  s.args().append("x", tbl()->lookup(node->x()));
+  s.state(locop::NodeSummary::State::Complete);
+  return true;
+}
+
 } // namespace
 
 namespace moco