From e8cc13e9335bb823e7b02f9f2cd588fdaaf49ed0 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Prasanna=20R/System=20SW=20/SRI-Bangalore/Engineer/?= =?utf8?q?=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 16 Oct 2018 06:55:41 +0530 Subject: [PATCH] [tfltool] OP selection: Enable for EqualOptions (#3112) This patch will enable EqualOptions in tfltool select_operator. Related issue: #2610 #2822 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 54586ae..ded6761 100755 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -513,6 +513,16 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t tflite.NegOptions.NegOptionsStart(new_builder) return tflite.NegOptions.NegOptionsEnd(new_builder) + # EqualOptions + import tflite.EqualOptions + if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().EqualOptions: + + equal_option = tflite.EqualOptions.EqualOptions() + equal_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) + + tflite.EqualOptions.EqualOptionsStart(new_builder) + return tflite.EqualOptions.EqualOptionsEnd(new_builder) + # PadV2Options: not supported # GreaterOptions: not supported # GreaterEqualOptions: not supported @@ -523,7 +533,6 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t # SparseToDenseOptions: not supported # TileOptions: not supported # ExpandDimsOptions: not supported - # EqualOptions: not supported # NotEqualOptions: not supported # ShapeOptions: not supported # PowOptions: not supported -- 2.7.4