torch-mlir/python
Gaurav Shukla 4b911ada40 [LINALG] Add E2E support for `aten.mean.dim` op
- This commit adds support for `aten.mean.dim` op.
- It also adds a new test script `stats.py` for statistics related ops.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
2022-05-04 20:11:42 +05:30
..
test Fix kwarg annotation in eager (#747) 2022-04-11 17:35:42 -05:00
torch_mlir Add tracing suport to `torch_mlir.compile`. 2022-05-03 09:08:40 -07:00
torch_mlir_e2e_test [LINALG] Add E2E support for `aten.mean.dim` op 2022-05-04 20:11:42 +05:30
CMakeLists.txt Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00