[all-commits] [llvm/llvm-project] 414709: [mlir][tosa] Add verifier for `ArgMax` operator (#...

[Georgios Pinitas via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 414709ee5c74ae35ac5eb204007fef35cc1c8079
      https://github.com/llvm/llvm-project/commit/414709ee5c74ae35ac5eb204007fef35cc1c8079
  Author: Georgios Pinitas <georgios.pinitas at arm.com>
  Date:   2023-10-09 (Mon, 09 Oct 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
    M mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
    M mlir/test/Dialect/Tosa/canonicalize.mlir
    M mlir/test/Dialect/Tosa/constrained_shapes.mlir
    M mlir/test/Dialect/Tosa/level_check.mlir

  Log Message:
  -----------
  [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 at arm.com>