[all-commits] [llvm/llvm-project] 28e642: [mlir][tosa] Add tosa.argmax to linalg lowering

[rsuderman via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 28e6420744f52cc39df0d0529c09385e61ddb8ef
      https://github.com/llvm/llvm-project/commit/28e6420744f52cc39df0d0529c09385e61ddb8ef
  Author: Rob Suderman <rob.suderman at gmail.com>
  Date:   2021-03-23 (Tue, 23 Mar 2021)

  Changed paths:
    M mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp
    M mlir/test/Conversion/TosaToLinalg/tosa-to-linalg.mlir

  Log Message:
  -----------
  [mlir][tosa] Add tosa.argmax to linalg lowering

Tosa's argmax lowering is representable as a linalg.indexed_generic
operation. Include the lowering to this type for both integer and
floating point types.

Differential Revision: https://reviews.llvm.org/D99137