torch-mlir/python/torch_mlir
Prashant Kumar b1a506624c Add decomposition of `aten.masked.tensor` op.
`aten.masked.tensor` op has been decomposed to `aten.masked.scalar` op.
2022-08-11 07:48:04 +05:30
..
_torch_mlir_custom_op_example Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc Resolve remaining LTC CI failures (#1110) 2022-07-30 09:40:02 -04:00
dialects Add decomposition of `aten.masked.tensor` op. 2022-08-11 07:48:04 +05:30
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py torch_mlir.compile: Allow ignoring traced shapes 2022-08-04 10:18:34 -07:00
compiler_utils.py Add named exception TorchMlirCompilerError. (#1064) 2022-07-15 16:32:36 -05:00