torch-mlir/e2e_testing/torchscript
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
..
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
basic.py Fix bug with transpose of negative dims 2021-10-25 15:50:55 -04:00
batchnorm.py E2e support for layernorm. 2021-10-04 14:15:13 -04:00
conv.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
elementwise.py Add aten::gelu lowering (#374) 2021-10-25 16:16:01 -07:00
main.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
quantized_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reduction.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
vision_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
xfail_sets.py Rename `tosa-to-linalg-on-tensors` to `tosa-to-linalg` 2021-10-25 20:43:54 +00:00