[Mlir-commits] [mlir] [mlir][linalg] Implement Conv2D using Winograd Conv2D algorithm (PR #96181)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Mon Jun 24 07:23:00 PDT 2024


https://github.com/Max191 requested changes to this pull request.

Nice improvements so far! A few comments:
1. The shapes for the filter transform op and the batch_matmul don't quite make sense to me. See the comments about the filter transform.
2. When the input image tensor is padded, the padded part of the tensor needs to be filled with zeros
3. It would be great to support NCHW conv layouts. This can come as a later PR, but would be really useful to have.

https://github.com/llvm/llvm-project/pull/96181


More information about the Mlir-commits mailing list