From f6673309097e42e7a151045801c00532b77a7463 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9C=A4=EC=A7=80=EC=98=81/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84?= =?utf8?q?=EC=9E=90?= Date: Tue, 16 Oct 2018 10:32:34 +0900 Subject: [PATCH] [tfltool] OP selector: Support more operator (#3094) This commit supports SQUEEZE option in op selector Signed-off-by: Jiyoung Yun --- tools/tflitefile_tool/select_operator.py | 23 ++++++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index ded6761..0c458cb 100755 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -440,7 +440,28 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t new_builder, div_option.FusedActivationFunction()) return tflite.DivOptions.DivOptionsEnd(new_builder) - # SqueezeOptions: not supported + # SqueezeOptions + import tflite.SqueezeOptions + if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().SqueezeOptions: + + squeeze_option = tflite.SqueezeOptions.SqueezeOptions() + squeeze_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) + + squeeze_dims_num = squeeze_option.SqueezeDimsLength() + if squeeze_dims_num != 0: + tflite.SqueezeOptions.SqueezeOptionsStartSqueezeDimsVector( + new_builder, squeeze_dims_num) + for squeeze_dims_idx in reversed(range(squeeze_dims_num)): + squeeze_dims_val = squeeze_option.SqueezeDims(squeeze_dims_idx) + new_builder.PrependInt32(squeeze_dims_val) + new_squeeze_dims = new_builder.EndVector(squeeze_dims_num) + + tflite.SqueezeOptions.SqueezeOptionsStart(new_builder) + if squeeze_dims_num != 0: + tflite.SqueezeOptions.SqueezeOptionsAddSqueezeDims(new_builder, + new_squeeze_dims) + return tflite.SqueezeOptions.SqueezeOptionsEnd(new_builder) + # SequenceRNNOptions: not supported # StridedSliceOptions -- 2.7.4