[tflchef] Add ReLU6 in activation (#2312)

This commit will add ReLU6 in activation

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

diff --git a/contrib/tflchef/core/src/Convert.cpp b/contrib/tflchef/core/src/Convert.cpp
index 6b0d3e9..7adddef 100644 (file)
--- a/contrib/tflchef/core/src/Convert.cpp
+++ b/contrib/tflchef/core/src/Convert.cpp
@@ -41,6 +41,8 @@ tflite::ActivationFunctionType as_tflite_activation(const tflchef::Activation &v
     return tflite::ActivationFunctionType_NONE;
   case tflchef::RELU:
     return tflite::ActivationFunctionType_RELU;
+  case tflchef::RELU6:
+    return tflite::ActivationFunctionType_RELU6;
   default:
     break;
   }

diff --git a/contrib/tflchef/proto/tflchef.proto b/contrib/tflchef/proto/tflchef.proto
index 77c2cb2..af1d253 100644 (file)
--- a/contrib/tflchef/proto/tflchef.proto
+++ b/contrib/tflchef/proto/tflchef.proto
@@ -34,6 +34,7 @@ enum Padding {
 enum Activation {
   NONE = 0;
   RELU = 1;
+  RELU6 = 3;
 }
 
 message Conv2DOptions