From c84921ec5c0ecc33bde4d5dadf422de780762132 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: Wed, 5 Dec 2018 05:51:42 +0530 Subject: [PATCH] [tfltool] Enable select operator for LogicalNot op (#3819) This patch enables LogicalNot in tflitetool in select_operator.py Signed-off-by: prasannar --- tools/tflitefile_tool/select_operator.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) mode change 100755 => 100644 tools/tflitefile_tool/select_operator.py diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py old mode 100755 new mode 100644 index 3469041..0100b64 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -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 -- 2.7.4