From: Prasanna R/SNAP /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Date: Wed, 5 Dec 2018 00:21:42 +0000 (+0530)
Subject: [tfltool] Enable select operator for LogicalNot op (#3819)
X-Git-Tag: 0.3~210
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=c84921ec5c0ecc33bde4d5dadf422de780762132;p=platform%2Fcore%2Fml%2Fnnfw.git

[tfltool] Enable select operator for LogicalNot op (#3819)

This patch enables LogicalNot in tflitetool in select_operator.py

Signed-off-by: prasannar <prasanna.r@samsung.com>
---

diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py
old mode 100755
new mode 100644
index 3469041..0100b64
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -730,7 +730,16 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
         return tflite.LogicalOrOptions.LogicalOrOptionsEnd(new_builder)
 
     # OneHotOptions: not supported
-    # LogicalNotOptions: not supported
+
+    # LogicalNotOptions
+    import tflite.LogicalNotOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().LogicalNotOptions:
+
+        equal_option = tflite.LogicalNotOptions.LogicalNotOptions()
+        equal_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.LogicalNotOptions.LogicalNotOptionsStart(new_builder)
+        return tflite.LogicalNotOptions.LogicalNotOptionsEnd(new_builder)
 
     # UnpackOptions:
     import tflite.UnpackOptions