90f0d7f935
This pattern only works for normal convolutions. It does not work for depthwise convolutions. The Linalg conv ops are defined with static rank, so it only supports 1d/2d/3d cases, which are the most typical cases. This also refactors out the same check in lmhlo.conv lowering. PiperOrigin-RevId: 359503527 |
||
---|---|---|
.. | ||
IR | ||
transforms | ||
CMakeLists.txt |