From 19f26e015221aaeb1ce9e00798ccfa1260a2dd5b Mon Sep 17 00:00:00 2001 From: =?utf8?q?Prasanna=20R/SNAP=20/SRI-Bangalore/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 4 Dec 2018 06:27:08 +0530 Subject: [PATCH] [tfltool] Enable select operator for Dequantize op (#3818) This patch enables Dequantize in tflitetool in select_operator.py Signed-off-by: prasannar --- tools/tflitefile_tool/select_operator.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) 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 -- 2.7.4