mlir-hlo/lib/Dialect/mhlo
Hanhan Wang 90f0d7f935 Add support for lowering mhlo.conv to Linalg on tensors.
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
2021-02-25 05:59:08 -08:00
..
IR [MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. 2021-02-22 08:43:00 -08:00
transforms Add support for lowering mhlo.conv to Linalg on tensors. 2021-02-25 05:59:08 -08:00
CMakeLists.txt Add CMake files and lit configurations, enough for `ninja check-mlir-hlo` to pass on all the tests 2020-08-07 22:14:34 -07:00