llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python/torch_mlir_e2e_test
History
Ramiro Leal-Cavazos a7af1fd873
Add support for `dim=None` to `AtenMeanDimOp` (#1129) 
PyTorch recently added support for `dim=None` in the `torch.mean`
op (2bfae07a79). This
commit adds the corresponding support in torch-mlir.
 		2022-08-02 16:08:06 +00:00
..
eager_backends add ci tests (#754) 2022-05-25 14:59:59 -05:00
linalg_on_tensors_backends add e2e support for aten.atan2 (#1117) 2022-08-02 11:39:41 -04:00
test_suite Add support for `dim=None` to `AtenMeanDimOp` (#1129) 2022-08-02 16:08:06 +00:00
torchscript Reference Lazy Backend (#1045) 2022-07-30 09:40:02 -04:00
tosa_backends bump llvm tag to 3580daa (#1078) 2022-07-18 16:49:03 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
__init__.py Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
utils.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00