[mlir][tosa] Separate tosa.transpose_conv decomposition and added stride support
authorRob Suderman <rob.suderman@gmail.com>
Tue, 23 Nov 2021 03:43:06 +0000 (19:43 -0800)
committerRob Suderman <rob.suderman@gmail.com>
Tue, 23 Nov 2021 20:16:44 +0000 (12:16 -0800)
commit54eec7cafc396f3d1444aacf4f1ed71fceb4e503
tree617d2829fe0fc0f221344b6e154e43f2c3d34111
parent44bb69b1e012bdc905aa679008ee7ed3d71061b1
[mlir][tosa] Separate tosa.transpose_conv decomposition and added stride support

Transpose convolution decomposition is now performed in a separate pass. This
allows padding / constant propagation to be performed at the TOSA level. It
also adds support for striding when there is no dilation.

Differential Revision: https://reviews.llvm.org/D114409
mlir/include/mlir/Dialect/Tosa/Transforms/Passes.h
mlir/include/mlir/Dialect/Tosa/Transforms/Passes.td
mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp
mlir/lib/Conversion/TosaToLinalg/TosaToLinalgPass.cpp
mlir/lib/Dialect/Tosa/Transforms/CMakeLists.txt
mlir/lib/Dialect/Tosa/Transforms/TosaDecomposeTransposeConv.cpp [new file with mode: 0644]
mlir/test/Conversion/TosaToLinalg/tosa-to-linalg.mlir
mlir/test/Dialect/Tosa/tosa-decompose-transpose-conv.mlir [new file with mode: 0644]