[all-commits] [llvm/llvm-project] f2832c: [mlir][tosa] Added shape propagation for TOSA pool...

[rsuderman via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f2832c2295c6076b51a35d0d7b304c08e1b41c29
      https://github.com/llvm/llvm-project/commit/f2832c2295c6076b51a35d0d7b304c08e1b41c29
  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
    M mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir

  Log Message:
  -----------
  [mlir][tosa] Added shape propagation for TOSA pool operations.

Pool operations perform the same shape propagation. Included the shape
propagation and tests for these avg_pool2d and max_pool2d.

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