MLBEDSW-7652: Add mean support for batch and channel when shape is 1 (1d5e8599) · Commits · artificial-intelligence / ethos-u / Vela

Commit 1d5e8599 authored Jun 27, 2023 by Alexander Hansson Committed by Johan Alfvén Jul 11, 2023

MLBEDSW-7652: Add mean support for batch and channel when shape is 1



- Add support for batch and depth channels when shape is 1
- Refactor reshaping in convert_mean_to_depthwise_conv

Signed-off-by: Alexander Hansson <Alexander.Hansson@arm.com>
Change-Id: If663395934ab58c76ba92b6ebaaf484a389ae699

parent ca9cc420

Hide whitespace changes

Inline Side-by-side

Please register or to comment