[mlir][tosa] Migrate tosa to more efficient linalg.conv
authorRob Suderman <rob.suderman@gmail.com>
Wed, 11 Aug 2021 18:05:08 +0000 (11:05 -0700)
committerRob Suderman <rob.suderman@gmail.com>
Wed, 11 Aug 2021 18:05:12 +0000 (11:05 -0700)
commit7de439b2be4a046da541b625812f2fe34c54c4b9
treee8ae474a563f24c8f2ea295d7de4fbbe979b95d6
parentc1a8f12873783e8f4827437f6b2dddadfc58109d
[mlir][tosa] Migrate tosa to more efficient linalg.conv

Existing linalg.conv2d is not well optimized for performance. Changed to a
version that is more aligned for optimziation. Include the corresponding
transposes to use this optimized version.

This also splits the conv and depthwise conv into separate implementations
to avoid overly complex lowerings.

Reviewed By: antiagainst

Differential Revision: https://reviews.llvm.org/D107504
mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp
mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
mlir/test/Conversion/TosaToLinalg/tosa-to-linalg.mlir
mlir/test/Dialect/Linalg/named-ops.mlir