[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:
Georgios Pinitas <georgios.pinitas@arm.com>
Loading
Please sign in to comment