[moco-tf] Knobs to import as TFRelu and TFRelu6 (#6181)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Sun, 4 Aug 2019 23:40:29 +0000 (08:40 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Sun, 4 Aug 2019 23:40:29 +0000 (08:40 +0900)
This will add two knobs to import as TFRelu and TFRelu6

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
compiler/moco-tf/src/Knob.lst

index 578ec4a..74f05f8 100644 (file)
@@ -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)