commit 4f728c04bcc90742d9d57b0e253be68a7251984f
author Dwight Lidman <dwight.lidman@arm.com> Thu Dec 17 15:14:45 2020 +0100
committer Louis Verhaard <louis.verhaard@arm.com> Thu Feb 25 17:27:27 2021 +0100
tree 7e205e9fc90a36606d1939af406af20fa10b448d
parent 0cbb166cd032c779bd4681afef8097f0831ac8be

MLBEDSW-1499: Add MEAN operator

This commit adds support for the MEAN operator,
with some caveats.

Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: I165cb26cb5aefd68e70d2cfc68291ccf7b778921

ethosu/vela/tflite_mapping.py

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index 41d57c0..b526ec5 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -563,7 +563,7 @@
     BuiltinOperator.BATCH_TO_SPACE_ND: (Op.BatchToSpaceND, OptionsSerializer("BatchToSpaceNDOptions")),
     BuiltinOperator.SPACE_TO_BATCH_ND: (Op.SpaceToBatchND, OptionsSerializer("SpaceToBatchNDOptions")),
     BuiltinOperator.TRANSPOSE: (Op.Transpose, OptionsSerializer("TransposeOptions")),
-    BuiltinOperator.MEAN: (Op.Mean, None),
+    BuiltinOperator.MEAN: (Op.Mean, reducer_opts),
     BuiltinOperator.SUB: (Op.Sub, OptionsSerializer("SubOptions", (fused_act, "pot_scale_int16",))),
     BuiltinOperator.DIV: (Op.Div, OptionsSerializer("DivOptions", (fused_act,))),
     BuiltinOperator.SQUEEZE: (Op.Squeeze, OptionsSerializer("SqueezeOptions", (("squeeze_dims", is_int_vec),))),