[mlir][Linalg] Allow tiling of batch dimension for convolution ops with padding.
authorMaheshRavishankar <ravishankarm@google.com>
Tue, 31 Mar 2020 15:59:45 +0000 (08:59 -0700)
committerMaheshRavishankar <ravishankarm@google.com>
Tue, 31 Mar 2020 16:22:38 +0000 (09:22 -0700)
commitda7b6fe942fc9bbf535321477d614051d05aee4b
tree18ab6fbb1e646cfcfd4fd7add3bfabc947f154ef
parentb0cd7b2799ebcac0e456abf60cea2ca421a83945
[mlir][Linalg] Allow tiling of batch dimension for convolution ops with padding.

Existing tiling implementation of Linalg would still work for tiling
the batch dimensions of the convolution op.

Differential Revision: https://reviews.llvm.org/D76637
mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
mlir/test/Dialect/Linalg/tile_conv_padding.mlir [new file with mode: 0644]