torch-mlir/lib
Vivek Khandelwal a11ef674a7 [MLIR][TORCH] Add E2E support for aten.baddbmm op
This commit decomposes `aten.baddbmm` op into `aten.bmm`,
`aten.mul.Scalar`, and `aten.add.Tensor` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-06-07 22:26:28 +05:30
..
CAPI Add tracing suport to `torch_mlir.compile`. 2022-05-03 09:08:40 -07:00
Conversion Add E2E support for AtenLogicalOrOp. (#883) 2022-06-03 16:21:03 -07:00
Dialect [MLIR][TORCH] Add E2E support for aten.baddbmm op 2022-06-07 22:26:28 +05:30
RefBackend llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
CMakeLists.txt Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00
InitAll.cpp Add TMTensor dialect to torch-mlir 2022-02-15 16:45:38 -05:00