From 2685b7aa01018fa9508f535a11f1cdf085d81c06 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=82=A8=EA=B6=81=EC=84=9D/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?= =?utf8?q?=EC=9E=90?= Date: Fri, 16 Nov 2018 11:06:49 +0900 Subject: [PATCH] [tflchef] Add ReLU6 in activation (#2312) This commit will add ReLU6 in activation Signed-off-by: Seok NamKoong --- contrib/tflchef/core/src/Convert.cpp | 2 ++ contrib/tflchef/proto/tflchef.proto | 1 + 2 files changed, 3 insertions(+) diff --git a/contrib/tflchef/core/src/Convert.cpp b/contrib/tflchef/core/src/Convert.cpp index 6b0d3e9..7adddef 100644 --- 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 --- 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 -- 2.7.4