[moco-tf] Add TFFormattedGraph for TFSoftmax (#6647)

* [moco-tf] Add TFFormattedGraph for TFSoftmax

This commit will add `TFSoftmax` summary function in `TFFormattedGraph.cpp`

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

diff --git a/compiler/moco-tf/src/TFFormattedGraph.cpp b/compiler/moco-tf/src/TFFormattedGraph.cpp
index 9595270..fb64544 100644 (file)
--- a/compiler/moco-tf/src/TFFormattedGraph.cpp
+++ b/compiler/moco-tf/src/TFFormattedGraph.cpp
@@ -87,6 +87,7 @@ private:
   IMPLEMENT(TFReshape);
   IMPLEMENT(TFRsqrt);
   IMPLEMENT(TFShape);
+  IMPLEMENT(TFSoftmax);
   IMPLEMENT(TFSqueeze);
 #undef IMPLEMENT
 };
@@ -249,6 +250,14 @@ bool TFNodeSummaryBuilder::summary(const TFShape *node, locop::NodeSummary &s) c
   return true;
 }
 
+bool TFNodeSummaryBuilder::summary(const TFSoftmax *node, locop::NodeSummary &s) const
+{
+  s.opname("TF.Softmax");
+  s.args().append("logits", tbl()->lookup(node->logits()));
+  s.state(locop::NodeSummary::State::Complete);
+  return true;
+}
+
 bool TFNodeSummaryBuilder::summary(const TFSqueeze *node, locop::NodeSummary &s) const
 {
   s.opname("TF.Squeeze");