torch-mlir/python
Sean Silva e1b38e74dd Use upstream shape functions directly.
Now that upstream exposes them nicely, we can use them.

I noticed that we had added stuff into the upstream_shape_helpers.py
file (which was supposed to stay pristine), so some more shape functions
need to be upstreamed.

Going forward, all shape functions should be upstreamed similar to
https://github.com/pytorch/pytorch/pull/76889 instead of added in this
file.
2022-06-07 11:15:03 -07:00
..
test torch_mlir.compile: Fix API footgun 2022-06-05 18:10:07 -07:00
torch_mlir Use upstream shape functions directly. 2022-06-07 11:15:03 -07:00
torch_mlir_e2e_test [MLIR][TORCH] Add E2E support for aten.div.Tensor_mode op 2022-06-07 22:26:44 +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