[all-commits] [llvm/llvm-project] 98aad4: [tosa] Improve inferred shapes of TOSA operations

[Eric Kunze via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 98aad40806f18cd1d8f248d15efc75fe26840bd4
      https://github.com/llvm/llvm-project/commit/98aad40806f18cd1d8f248d15efc75fe26840bd4
  Author: Spenser Bauman <sbauman at mathworks.com>
  Date:   2023-05-25 (Thu, 25 May 2023)

  Changed paths:
    M mlir/lib/Dialect/Tosa/Transforms/TosaInferShapes.cpp
    M mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir

  Log Message:
  -----------
  [tosa] Improve inferred shapes of TOSA operations

The TosaInferShapes pass avoids updating the shapes of tensor operators
when the consumers are not TOSA operations, limiting the efficacy of
TosaInferShapes when the IR is a mix of TOSA and other operations.
This change attempts to update the result shapes when the consumers
themselves have reasonable type/shape inference methods.

Reviewed By: eric-k256

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