torch-mlir/lib
Prashant Kumar 5009cbf55c Add lowering of aten.matmul op.
Lowering of `aten.matmul` op is added from torch to linalg dialect.
The different cases correspond to
https://pytorch.org/docs/stable/generated/torch.matmul.html.
TODO: Broadcasting in case of batch-matmul is yet to be taken care of.

Signed-off-by: Prashant Kumar <prashant@nod-labs.com>
2021-10-26 12:45:09 -04:00
..
CAPI Bump llvm-project to f1b922188ead5ca492c8d8edd47921b013a22ae0. 2021-10-25 13:22:07 -07:00
Conversion Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
Dialect Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
RefBackend Add aten::gelu lowering (#374) 2021-10-25 16:16:01 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
InitAll.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00