From 3092b9f47d6587c3f840e40e0bb76c8b7a228fc3 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 08:40:29 +0900 Subject: [PATCH] [moco-tf] Knobs to import as TFRelu and TFRelu6 (#6181) This will add two knobs to import as TFRelu and TFRelu6 Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Knob.lst | 2 ++ 1 file changed, 2 insertions(+) diff --git a/compiler/moco-tf/src/Knob.lst b/compiler/moco-tf/src/Knob.lst index 578ec4a..74f05f8 100644 --- a/compiler/moco-tf/src/Knob.lst +++ b/compiler/moco-tf/src/Knob.lst @@ -11,6 +11,8 @@ KNOB_BOOL(ImportAsTFConst, true, Import Const node as TFConst node) KNOB_BOOL(ImportAsTFConv2D, true, Import Conv2D node as TFConv2D node) KNOB_BOOL(ImportAsTFIdentity, true, Import Identity node as TFIdentity node) KNOB_BOOL(ImportAsTFMaxPool, true, Import MaxPool node as TFMaxPool node) +KNOB_BOOL(ImportAsTFRelu, false, Import Relu node as TFRelu node) +KNOB_BOOL(ImportAsTFRelu6, false, Import Relu6 node as TFRelu6 node) // TensorFlow dialect transforms KNOB_BOOL(FuseBinaryIntoPreceding, true, Fuse Binary node to preceding node) -- 2.7.4