aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite
diff options
context:
space:
mode:
authorWilliam Isaksson <william.isaksson@arm.com>2023-07-17 13:03:09 +0000
committerFredrik Svedberg <fredrik.svedberg@arm.com>2023-07-24 13:09:58 +0000
commit2f9b6874a227d8fa056c2e2fd01e8c80824ee0bc (patch)
treeebde7518ef8380ce4e250610528b4030d140e12a /ethosu/vela/tflite
parent9cf63a3612491198a39f6bd1f4a587589b3ac20a (diff)
downloadethos-u-vela-2f9b6874a227d8fa056c2e2fd01e8c80824ee0bc.tar.gz
MLBEDSW-7165: Update to TensorFlow 2.12
- Updated FlatBuffers files using TensorFlow 2.12.0 schema - Added restriction for UnidirectionalSequenceLSTM to have 2D recurrent weights to handle that diagonal_recurrent_tensors attr is not currently supported. Change-Id: I104fd1f52485b9b83d644772dbcdeea2d17585f0 Signed-off-by: William Isaksson <william.isaksson@arm.com>
Diffstat (limited to 'ethosu/vela/tflite')
-rw-r--r--ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py12
1 files changed, 11 insertions, 1 deletions
diff --git a/ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py b/ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py
index 16641a74..73d298d2 100644
--- a/ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py
+++ b/ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py
@@ -63,7 +63,14 @@ class UnidirectionalSequenceLSTMOptions(object):
return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
return False
-def UnidirectionalSequenceLSTMOptionsStart(builder): builder.StartObject(5)
+ # UnidirectionalSequenceLSTMOptions
+ def DiagonalRecurrentTensors(self):
+ o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(14))
+ if o != 0:
+ return bool(self._tab.Get(flatbuffers.number_types.BoolFlags, o + self._tab.Pos))
+ return False
+
+def UnidirectionalSequenceLSTMOptionsStart(builder): builder.StartObject(6)
def Start(builder):
return UnidirectionalSequenceLSTMOptionsStart(builder)
def UnidirectionalSequenceLSTMOptionsAddFusedActivationFunction(builder, fusedActivationFunction): builder.PrependInt8Slot(0, fusedActivationFunction, 0)
@@ -81,6 +88,9 @@ def AddTimeMajor(builder, timeMajor):
def UnidirectionalSequenceLSTMOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs): builder.PrependBoolSlot(4, asymmetricQuantizeInputs, 0)
def AddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs):
return UnidirectionalSequenceLSTMOptionsAddAsymmetricQuantizeInputs(builder, asymmetricQuantizeInputs)
+def UnidirectionalSequenceLSTMOptionsAddDiagonalRecurrentTensors(builder, diagonalRecurrentTensors): builder.PrependBoolSlot(5, diagonalRecurrentTensors, 0)
+def AddDiagonalRecurrentTensors(builder, diagonalRecurrentTensors):
+ return UnidirectionalSequenceLSTMOptionsAddDiagonalRecurrentTensors(builder, diagonalRecurrentTensors)
def UnidirectionalSequenceLSTMOptionsEnd(builder): return builder.EndObject()
def End(builder):
return UnidirectionalSequenceLSTMOptionsEnd(builder) \ No newline at end of file