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/npu_performance.py

diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py
index 820c7d6..e315f1f 100644
--- a/ethosu/vela/npu_performance.py
+++ b/ethosu/vela/npu_performance.py
@@ -509,8 +509,7 @@
                 )
             else:
                 weight_tensor_shape = [
-                    primary_op.attrs["ksize"][1],
-                    primary_op.attrs["ksize"][2],
+                    *primary_op.get_kernel_size(),
                     1,
                     ifm_tensor_shape.depth,
                 ]