[all-commits] [llvm/llvm-project] 5a4e77: [mlir][tosa] Added more shape inference for tosa ops

[rsuderman via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 5a4e7760101581c394aa4235c55885ec960c8b3b
      https://github.com/llvm/llvm-project/commit/5a4e7760101581c394aa4235c55885ec960c8b3b
  Author: Rob Suderman <rob.suderman at gmail.com>
  Date:   2021-07-12 (Mon, 12 Jul 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
    M mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
    A mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir
    R mlir/test/Dialect/Tosa/tosa_infer_shapes.mlir

  Log Message:
  -----------
  [mlir][tosa] Added more shape inference for tosa ops

Added shape inference for:
- scatter
- gather
- transpose
- slice
- pad
- concat
- reduction operations

Also updated reshape for more aggressive shape inference.

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