diff options
Diffstat (limited to 'ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py')
-rw-r--r-- | ethosu/vela/tflite/UnidirectionalSequenceLSTMOptions.py | 12 |
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 |