Skip to content
Unverified Commit 414709ee authored by Georgios Pinitas's avatar Georgios Pinitas Committed by GitHub
Browse files

[mlir][tosa] Add verifier for `ArgMax` operator (#68410)



Verifier ensures that operator is valid by checking:

* Output type is I32
* Axis is within the rank of the tensor

Signed-off-by: default avatarGeorgios Pinitas <georgios.pinitas@arm.com>
parent b7188d28
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment