[moco-tf] Fix comments for Relu, Relu6 (#6202)

This will fix comments for Relu and Relu6 nodes

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

diff --git a/compiler/moco-tf/src/Op/Relu6.test.cpp b/compiler/moco-tf/src/Op/Relu6.test.cpp
index d0a4969..4d68323 100644 (file)
--- a/compiler/moco-tf/src/Op/Relu6.test.cpp
+++ b/compiler/moco-tf/src/Op/Relu6.test.cpp
@@ -101,7 +101,7 @@ TEST(TensorFlowImport, relu6_01)
   std::unique_ptr<loco::Graph> graph = importer.import(signature, graph_def);
 
   // what to test:
-  // - there should exist ReLU
+  // - there should exist ReLU6
   // - input node should not be nullptr
 
   loco::ReLU6 *relu6_node = moco::tf::test::find_first_node_bytype<loco::ReLU6>(graph.get());

diff --git a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp
index 8649981..491fd25 100644 (file)
--- a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp
+++ b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp
@@ -1067,14 +1067,14 @@ bool fix_shape(moco::tf::TFMul *node)
 
 bool fix_shape(moco::tf::TFRelu *node)
 {
-  // Output shape is same as the input
+  // Output shape is same as the features
   auto features = node->features();
   return copy_shapedata(features, node);
 }
 
 bool fix_shape(moco::tf::TFRelu6 *node)
 {
-  // Output shape is same as the input
+  // Output shape is same as the features
   auto features = node->features();
   return copy_shapedata(features, node);
 }