aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py')
-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