llvm-project/mlir/lib/Dialect/Tosa/IR
Rob Suderman f2832c2295 [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
2021-07-12 15:40:49 -07:00
..
TosaOps.cpp [mlir][tosa] Added shape propagation for TOSA pool operations. 2021-07-12 15:40:49 -07:00