diff options
author | William Isaksson <william.isaksson@arm.com> | 2023-07-17 13:03:09 +0000 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2023-07-24 13:09:58 +0000 |
commit | 2f9b6874a227d8fa056c2e2fd01e8c80824ee0bc (patch) | |
tree | ebde7518ef8380ce4e250610528b4030d140e12a /ethosu/vela/tflite_mapping.py | |
parent | 9cf63a3612491198a39f6bd1f4a587589b3ac20a (diff) | |
download | ethos-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_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index 83c55bb2..dba30c46 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -716,7 +716,14 @@ builtin_operator_map = { Op.UnidirectionalSequenceLstm, OptionsSerializer( "UnidirectionalSequenceLSTMOptions", - ("asymmetric_quantize_inputs", "cell_clip", fused_act, "proj_clip", "time_major"), + ( + "asymmetric_quantize_inputs", + "cell_clip", + "diagonal_recurrent_tensors", + fused_act, + "proj_clip", + "time_major", + ), ), TFLITE_IFM_WEIGHTS_INDICES, ), |