From f2f9bb1f3d4fad169732f820cba10615149e614c Mon Sep 17 00:00:00 2001 From: =?utf8?q?Devansh=20Bansal/SNAP=20/SRI-Bangalore/Engineer/=EC=82=BC?= =?utf8?q?=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Tue, 4 Dec 2018 16:31:36 +0530 Subject: [PATCH] [tfltool] Correction in select operator for ArgMax op (#3854) This patch removes "not supported" for ArgMax in tflitetool in select_operator.py Signed-off-by: b.devansh --- tools/tflitefile_tool/select_operator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index ca6b1f5..3469041 100755 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -640,8 +640,6 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t return tflite.DequantizeOptions.DequantizeOptionsEnd(new_builder) # MaximumMinimumOptions: not supported - # ArgMaxOptions: not supported - # LessOptions: not supported # ArgMaxOptions import tflite.ArgMaxOptions @@ -655,6 +653,8 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t arg_max_option.OutputType()) return tflite.ArgMaxOptions.ArgMaxOptionsEnd(new_builder) + # LessOptions: not supported + # NegOptions import tflite.NegOptions if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().NegOptions: -- 2.7.4