commit 42abec185a95ac8d5f2b32d541f587b3cd306e75
author Tim Hall <tim.hall@arm.com> Thu Feb 04 21:31:57 2021 +0000
committer Tim Hall <tim.hall@arm.com> Fri Feb 05 11:54:48 2021 +0000
tree 7c17eb47cb89a37b9f9d82add61385a5a82c40fe
parent 73e843f76dd71e4ab5e07a7616c2c4806ca6ac25

MLBEDSW-3771: Updated to TF 2.4 flatbuffer schema

Updated tflite loader and mappings from tensorflow 2.3 to tensorflow 2.4

Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I55884000ee139baf639bb0377008e0534f72fe94

ethosu/vela/tflite_mapping.py

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index 6fcbb95..c1e45c0 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -40,6 +40,7 @@
 from .tflite import ConcatenationOptions
 from .tflite import Conv2DOptions
 from .tflite import CosOptions
+from .tflite import CumsumOptions
 from .tflite import DensifyOptions
 from .tflite import DepthToSpaceOptions
 from .tflite import DepthwiseConv2DOptions
@@ -278,6 +279,7 @@
     BuiltinOptions.SelectV2Options: SelectV2Options.SelectV2Options,
     BuiltinOptions.WhileOptions: WhileOptions.WhileOptions,
     BuiltinOptions.BatchMatMulOptions: BatchMatMulOptions.BatchMatMulOptions,
+    BuiltinOptions.CumsumOptions: CumsumOptions.CumsumOptions,
 }
 
 builtin_options_inv_map = inverse_map(builtin_options_map)
@@ -690,6 +692,7 @@
     BuiltinOperator.DENSIFY: (Op.Densify, OptionsSerializer("DensifyOptions")),
     BuiltinOperator.SEGMENT_SUM: (Op.SegmentSum, OptionsSerializer("SegmentSumOptions")),
     BuiltinOperator.BATCH_MATMUL: (Op.BatchMatMul, OptionsSerializer("BatchMatMulOptions", ("adj_x", "adj_y"))),
+    BuiltinOperator.CUMSUM: (Op.Cumsum, OptionsSerializer("CumsumOptions", ("exclusive", "reverse"))),
     BuiltinOperator.CUSTOM: (Op.Custom, CustomOptionsSerializer()),
 }