diff options
Diffstat (limited to 'tools/tflitefile_tool/select_operator.py')
-rwxr-xr-x[-rw-r--r--] | tools/tflitefile_tool/select_operator.py | 27 |
1 files changed, 23 insertions, 4 deletions
diff --git a/tools/tflitefile_tool/select_operator.py b/tools/tflitefile_tool/select_operator.py index 4dd516398..b122b62e9 100644..100755 --- a/tools/tflitefile_tool/select_operator.py +++ b/tools/tflitefile_tool/select_operator.py @@ -184,7 +184,7 @@ def GenerateTensor(new_builder, selected_tensor, used_buffers_dic): # Create quantization quantization = selected_tensor.Quantization() - if quantization != 0: + if quantization != None: new_quantization = GenerateQuantization(new_builder, quantization) # Create tensor @@ -194,7 +194,7 @@ def GenerateTensor(new_builder, selected_tensor, used_buffers_dic): tflite.Tensor.TensorAddBuffer(new_builder, new_buffer_idx) if name_string != "": tflite.Tensor.TensorAddName(new_builder, new_name) - if quantization != 0: + if quantization != None: tflite.Tensor.TensorAddQuantization(new_builder, new_quantization) return tflite.Tensor.TensorEnd(new_builder) @@ -709,7 +709,16 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t # SparseToDenseOptions: not supported # TileOptions: not supported - # ExpandDimsOptions: not supported + + # ExpandDimsOptions: + import tflite.ExpandDimsOptions + if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().ExpandDimsOptions: + + expanddims_option = tflite.ExpandDimsOptions.ExpandDimsOptions() + expanddims_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) + + tflite.ExpandDimsOptions.ExpandDimsOptionsStart(new_builder) + return tflite.ExpandDimsOptions.ExpandDimsOptionsEnd(new_builder) # NotEqualOptions: import tflite.NotEqualOptions @@ -721,7 +730,17 @@ def GenerateBuiltinOption(new_builder, selected_builtin_option, builtin_option_t tflite.NotEqualOptions.NotEqualOptionsStart(new_builder) return tflite.NotEqualOptions.NotEqualOptionsEnd(new_builder) - # ShapeOptions: not supported + # ShapeOptions: + import tflite.ShapeOptions + if builtin_option_type == tflite.BuiltinOptions.BuiltinOptions().ShapeOptions: + + shape_option = tflite.ShapeOptions.ShapeOptions() + shape_option.Init(selected_builtin_option.Bytes, selected_builtin_option.Pos) + + tflite.ShapeOptions.ShapeOptionsStart(new_builder) + tflite.ShapeOptions.ShapeOptionsAddOutType(new_builder, shape_option.OutType()) + return tflite.ShapeOptions.ShapeOptionsEnd(new_builder) + # PowOptions: not supported # ArgMinOptions: not supported # FakeQuantOptions: not supported |