MLBEDSW-10635: ArgMax support for Ethos-U55 / Ethos-U65 (Regor) (TOSA/TFLite) (!155) · Merge requests · artificial-intelligence / ethos-u / Vela

Max Bergfelt requested to merge MLBEDSW-10635 into main Apr 07, 2025

Ported ArgMax support from Vela Python to Regor where ArgMax is rewritten as a series of other operators (DepthwiseConv2D -> MaxPool -> LUT -> Cast).

Change-Id: Ia3ee931b6a07750ea73d273bfec7fe2d791de3fd Signed-off-by: Max Bergfelt max.bergfelt@arm.com

MLBEDSW-10635: ArgMax support for Ethos-U55 / Ethos-U65 (Regor) (TOSA/TFLite)