mirror of https://github.com/llvm/torch-mlir
5009cbf55c
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> |
||
---|---|---|
.. | ||
argmax.py | ||
basic.py | ||
batchnorm.py | ||
conv.py | ||
elementwise.py | ||
main.py | ||
matmul.py | ||
mlp.py | ||
quantized_models.py | ||
reduction.py | ||
vision_models.py | ||
xfail_sets.py |