From 50ace7e3e82bdc5ea88bf79076f854a98d18d88f 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:29:51 +0530 Subject: [PATCH] [tfltool] Enable select operator for ArgMax op (#3813) This patch will enable ArgMax op in tflitetool in select_operator.py Signed-off-by: prasannar --- tools/tflitefile_tool/select_operator.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index 42854c2..48d8145 100755 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -643,6 +643,18 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t # ArgMaxOptions: not supported # LessOptions: not supported + # ArgMaxOptions + import tflite.ArgMaxOptions + if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().ArgMaxOptions: + + arg_max_option = tflite.ArgMaxOptions.ArgMaxOptions() + arg_max_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) + + tflite.ArgMaxOptions.ArgMaxOptionsStart(new_builder) + tflite.ArgMaxOptions.ArgMaxOptionsAddOutputType(new_builder, + arg_max_option.OutputType()) + return tflite.ArgMaxOptions.ArgMaxOptionsEnd(new_builder) + # NegOptions import tflite.NegOptions if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().NegOptions: -- 2.7.4