[mlir][tosa] Added shape propagation for TOSA pool operations.
authorRob Suderman <rob.suderman@gmail.com>
Thu, 8 Jul 2021 22:17:15 +0000 (15:17 -0700)
committerRob Suderman <rob.suderman@gmail.com>
Mon, 12 Jul 2021 22:40:49 +0000 (15:40 -0700)
commitf2832c2295c6076b51a35d0d7b304c08e1b41c29
treee335f96c2d979b604e81bacbeb5bcba920e120c8
parent6611fbc62af59a1d28a9f310d2e95267911d4385
[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
mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td
mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
mlir/test/Dialect/Tosa/tosa-infer-shapes.mlir