diff options
author | William Isaksson <william.isaksson@arm.com> | 2023-11-22 22:27:58 +0100 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-12-20 08:46:09 +0000 |
commit | f4a511ffe7cd6c77e2effcbdf0843b2ef89d8df4 (patch) | |
tree | 9edfe95c8fa72bf8178e478a45cd69a8dc0e8ee4 /ethosu/vela/tflite_writer.py | |
parent | d2e03c61b028f9f84fd61812ea8905bf68f482e3 (diff) | |
download | ethos-u-vela-f4a511ffe7cd6c77e2effcbdf0843b2ef89d8df4.tar.gz |
MLBEDSW-8157: Update to TensorFlow 2.15
Updates to TensorFlow 2.15. No StableHLO operators were added to Vela since these are subject to change and have almost no runtime support.
- FlatBuffers version was unchanged.
Change-Id: I9a506a2dcc2e0bc2498742e857bbb6d69b19ac1b
Signed-off-by: William Isaksson <william.isaksson@arm.com>
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Diffstat (limited to 'ethosu/vela/tflite_writer.py')
-rw-r--r-- | ethosu/vela/tflite_writer.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index 6d85567c..44ce711f 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -343,8 +343,12 @@ class TFLiteSerialiser: Operator.OperatorAddIntermediates(builder, intermediates_offset) if builtin_opt_offset is not None: - Operator.OperatorAddBuiltinOptionsType(builder, opt_serializer.builtin_opt_type) - Operator.OperatorAddBuiltinOptions(builder, builtin_opt_offset) + if opt_serializer.builtin_opt_type < 127: + Operator.OperatorAddBuiltinOptionsType(builder, opt_serializer.builtin_opt_type) + Operator.OperatorAddBuiltinOptions(builder, builtin_opt_offset) + else: + Operator.OperatorAddBuiltinOptions2Type(builder, opt_serializer.builtin_opt_type % 127) + Operator.OperatorAddBuiltinOptions2(builder, builtin_opt_offset) if custom_opt_offset is not None: Operator.OperatorAddCustomOptions(builder, custom_opt_offset) Operator.OperatorAddCustomOptionsFormat(builder, opt_serializer.custom_opt_format) |