torch-mlir/frontends/pytorch/examples
Sean Silva 1dfcfa9cd1 Add aten.mm op and "test" it e2e.
Note that unlike aten.matmul which has dynamic behavior
depending on the argument ranks (can do matrix-matrix, matrix-vector,
batch matmul, etc.), aten.mm is just a vanilla matrix
multiply, which can be lowered precisely to tcf.matmul.

The "test" is really just an example that I stared at while getting my
feet wet with this. We probably want something that actually tests this
as part of `ninja check-npcomp`.
2020-11-20 17:21:24 -08:00
..
mm_e2e.py Add aten.mm op and "test" it e2e. 2020-11-20 17:21:24 -08:00
mul_maximum_e2e.py Repurpose numpy-compiler compiler/runtime flow for PyTorch. 2020-11-11 10:38:13 -08:00