[tfltool] Enable select operator for Dequantize op (#3818)
authorPrasanna R/SNAP /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Tue, 4 Dec 2018 00:57:08 +0000 (06:27 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Tue, 4 Dec 2018 00:57:08 +0000 (09:57 +0900)
This patch enables Dequantize in tflitetool in select_operator.py

Signed-off-by: prasannar <prasanna.r@samsung.com>
tools/tflitefile_tool/select_operator.py

index 86c55a3..192ec18 100755 (executable)
@@ -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