[exo] adding TFLRelu6Converter into convert.cpp (#8253)

This adds TFLRelu6Converter into convert.cpp with related knob on.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>

diff --git a/compiler/exo/src/Conversions.h b/compiler/exo/src/Conversions.h
index 1d969e6..370665e 100644 (file)
--- a/compiler/exo/src/Conversions.h
+++ b/compiler/exo/src/Conversions.h
@@ -31,7 +31,7 @@
 // TODO loco::FixedReshape
 #include "Conversion/MaxPool2DConverter.h"
 #include "Conversion/ReluConverter.h"
-// TODO loco::ReLU6
+#include "Conversion/Relu6Converter.h"
 // TODO loco::Tanh
 #include "Conversion/TensorConcatConverter.h"
 // TODO loco::TensorBiasAdd

diff --git a/compiler/exo/src/Convert.cpp b/compiler/exo/src/Convert.cpp
index 9ce0f9a..6c5236e 100644 (file)
--- a/compiler/exo/src/Convert.cpp
+++ b/compiler/exo/src/Convert.cpp
@@ -89,7 +89,9 @@ void convert_to_TFLNodes(loco::Graph *graph)
     if (get<Knob::ConvertRelu>())
       phase.emplace_back(stdex::make_unique<ReluConverter>());
 
-    // TODO loco::ReLU6
+    if (get<Knob::ConvertRelu6>())
+      phase.emplace_back(stdex::make_unique<Relu6Converter>());
+
     // TODO loco::Tanh
 
     if (get<Knob::ConvertTensorConcat>())

diff --git a/compiler/exo/src/Knob.lst b/compiler/exo/src/Knob.lst
index 213c23c..75ae300 100644 (file)
--- a/compiler/exo/src/Knob.lst
+++ b/compiler/exo/src/Knob.lst
@@ -23,7 +23,7 @@ KNOB_BOOL(ConvertFeatureBiasAdd, false, Convert loco::FeatureBiasAdd to TFLAdd)
 // TODO loco::FixedReshape
 KNOB_BOOL(ConvertMaxPool2D, true, Convert loco::MaxPool2D to TFLMaxPool2D)
 KNOB_BOOL(ConvertRelu, true, Convert loco::Relu to TFLRelu)
-// TODO loco::ReLU6
+KNOB_BOOL(ConvertRelu6, true, Convert loco::Relu6 to TFLRelu6)
 // TODO loco::Tanh
 KNOB_BOOL(ConvertTensorConcat, false, Convert loco::TensorConcat to TFLConcatenate)
 // TODO loco::TensorBiasAdd