torch-mlir/python/torch_mlir
Tanyo Kwok d4f1f41435
[MLIR][TORCH] Add decomposition of aten.repeat (#932)
* [MLIR][TORCH] Add decomposition of aten.repeat

* refine & rebase

* refine static shapes

* add e2e test

* Rebase and Refine naming style
2022-07-01 13:02:31 +08:00
..
_torch_mlir_custom_op_example Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
dialects [MLIR][TORCH] Add decomposition of aten.repeat (#932) 2022-07-01 13:02:31 +08:00
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py torch_mlir.compile: Fix API footgun 2022-06-05 18:10:07 -07:00
compiler_utils.py Update debug options in compilation errors (#913) 2022-06-07 10:55:54 -07:00