From: Prasanna R/SNAP /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Date: Tue, 4 Dec 2018 00:57:08 +0000 (+0530)
Subject: [tfltool] Enable select operator for Dequantize op (#3818)
X-Git-Tag: 0.3~245
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=19f26e015221aaeb1ce9e00798ccfa1260a2dd5b;p=platform%2Fcore%2Fml%2Fnnfw.git

[tfltool] Enable select operator for Dequantize op (#3818)

This patch enables Dequantize 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
index 86c55a3..192ec18 100755
--- a/tools/tflitefile_tool/select_operator.py
+++ b/tools/tflitefile_tool/select_operator.py
@@ -629,7 +629,16 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t
         tflite.CastOptions.CastOptionsStart(new_builder)
         return tflite.CastOptions.CastOptionsEnd(new_builder)
 
-    # DequantizeOptions: not supported
+    # DequantizeOptions:
+    import tflite.DequantizeOptions
+    if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().DequantizeOptions:
+
+        dequantize_option = tflite.DequantizeOptions.DequantizeOptions()
+        dequantize_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos)
+
+        tflite.EqualOptions.DequantizeOptionsStart(new_builder)
+        return tflite.DequantizeOptions.DequantizeOptionsEnd(new_builder)
+
     # MaximumMinimumOptions: not supported
     # ArgMaxOptions: not supported
     # LessOptions: not supported