tree c5cfef45fd915dc44b313f5fd8510f375dafa806
parent 3340a88d51cdce4d00ea82a2211d9e7fd40cf5c5
author Rickard Bolin <rickard.bolin@arm.com> 1671453220 +0000
committer Rickard Bolin <rickard.bolin@arm.com> 1679479881 +0000

MLBEDSW-6435: Implement support for ArgMax along depth dimension

- Add support for ArgMax along depth dimension with a depth limit of 127.
- Only supports 8-bit input and 32-bit output

Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: I5f6f0503135bebabbb1ca637f9729587b7c60740