From 2b7cc1e14aafe6985fb25fd6c1a7f823999c71ef Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 5 Aug 2019 16:12:01 +0900 Subject: [PATCH] [moco-tf] Fix comments for Relu, Relu6 (#6202) This will fix comments for Relu and Relu6 nodes Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Op/Relu6.test.cpp | 2 +- compiler/moco-tf/src/Transforms/FixShapeTransform.cpp | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/compiler/moco-tf/src/Op/Relu6.test.cpp b/compiler/moco-tf/src/Op/Relu6.test.cpp index d0a4969..4d68323 100644 --- 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 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(graph.get()); diff --git a/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp b/compiler/moco-tf/src/Transforms/FixShapeTransform.cpp index 8649981..491fd25 100644 --- 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); } -- 2.7.4