[mlir][linalg] Prepare drop unit dims for scalar operands.
authorTobias Gysi <gysit@google.com>
Fri, 11 Jun 2021 12:53:21 +0000 (12:53 +0000)
committerTobias Gysi <gysit@google.com>
Fri, 11 Jun 2021 13:18:06 +0000 (13:18 +0000)
commitf6b4e081dc9cf74fb5c22439f552fa035f2c2651
tree753892ffac6c2b0ab9769c4826b1bee7734e6bd6
parent4125b46232caf6a558470670f87b56ccb46d61aa
[mlir][linalg] Prepare drop unit dims for scalar operands.

Adapt drop unit dims for structured ops taking scalar operands.

Differential Revision: https://reviews.llvm.org/D103890
mlir/lib/Dialect/Linalg/Transforms/DropUnitDims.cpp