From: Shubham Gupta/SNAP /SRI-Bangalore/Engineer/삼성전자 <shub98.gupta@samsung.com>
Date: Tue, 4 Dec 2018 01:04:24 +0000 (+0530)
Subject: [tfltool] Enable select operator for LogicalOr op (#3824)
X-Git-Tag: 0.3~239
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=a6200cbde8261506bbc4455a8423af0f4fe1781a;p=platform%2Fcore%2Fml%2Fnnfw.git

[tfltool] Enable select operator for LogicalOr op (#3824)

This patch will enalbe select operator for Logicalor op

Signed-off-by: shubham <shub98.gupta@samsung.com>
---

diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py
index fc570d8..c037472 100755
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -718,7 +718,17 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
     # ArgMinOptions: not supported
     # FakeQuantOptions: not supported
     # PackOptions: not supported
-    # LogicalOrOptions: not supported
+
+    # LogicalOrOptions:
+    import tflite.LogicalOrOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().LogicalOrOptions:
+
+        logical_or_option = tflite.LogicalAndOptions.LogicalOrOptions()
+        logical_or_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.LogicalOrOptions.LogicalOrOptionsStart(new_builder)
+        return tflite.LogicalOrOptions.LogicalOrOptionsEnd(new_builder)
+
     # OneHotOptions: not supported
     # LogicalNotOptions: not supported