[tfltool] Enable select operator for LogicalNot op (#3819)
authorPrasanna R/SNAP /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Wed, 5 Dec 2018 00:21:42 +0000 (05:51 +0530)
committer박세희/동작제어Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Wed, 5 Dec 2018 00:21:42 +0000 (09:21 +0900)
This patch enables LogicalNot in tflitetool in select_operator.py

Signed-off-by: prasannar <prasanna.r@samsung.com>
tools/tflitefile_tool/select_operator.py [changed mode: 0755->0644]

old mode 100755 (executable)
new mode 100644 (file)
index 3469041..0100b64
@@ -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