tree 3657084dd74701207654e673b82dab3793afdcc3
parent 2f75457df27da610afdf01b1c86535030b022a45
author Dwight Lidman <dwight.lidman@arm.com> 1619600146 +0200
committer Dwight Lidman <dwight.lidman@arm.com> 1619682926 +0000

MLBEDSW-4501: Support MEAN single axis variation

When a MEAN operator with a single reduction axis
specifies the axis index attribute as an array with
a single element rather than a scalar index, the
operator is placed on the CPU even though it is
technically supported.
This commit fixes this issue and also adds some new
tests for the axis constraints.

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