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/FullyConnectedOptions.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/FullyConnectedOptions.py')
-rw-r--r-- | ethosu/vela/tflite/FullyConnectedOptions.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/ethosu/vela/tflite/FullyConnectedOptions.py b/ethosu/vela/tflite/FullyConnectedOptions.py index 67ca53c4..1ac220c7 100644 --- a/ethosu/vela/tflite/FullyConnectedOptions.py +++ b/ethosu/vela/tflite/FullyConnectedOptions.py @@ -56,7 +56,14 @@ class FullyConnectedOptions(object): return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos)) return False -def FullyConnectedOptionsStart(builder): builder.StartObject(4) + # FullyConnectedOptions + def QuantizedBiasType(self): + o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(12)) + if o != 0: + return self._tab.Get(flatbuffers.number_types.Int8Flags, o + self._tab.Pos) + return 0 + +def FullyConnectedOptionsStart(builder): builder.StartObject(5) def Start(builder): return FullyConnectedOptionsStart(builder) def FullyConnectedOptionsAddFusedActivationFunction(builder, fusedActivationFunction): builder.PrependInt8Slot(0, fusedActivationFunction, 0) @@ -71,6 +78,9 @@ def AddKeepNumDims(builder, keepNumDims): def FullyConnectedOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs): builder.PrependBoolSlot(3, asymmetricQuantizeInputs, 0) def AddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs): return FullyConnectedOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs) +def FullyConnectedOptionsAddQuantizedBiasType(builder, quantizedBiasType): builder.PrependInt8Slot(4, quantizedBiasType, 0) +def AddQuantizedBiasType(builder, quantizedBiasType): + return FullyConnectedOptionsAddQuantizedBiasType(builder, quantizedBiasType) def FullyConnectedOptionsEnd(builder): return builder.EndObject() def End(builder): return FullyConnectedOptionsEnd(builder)
\ No newline at end of file |