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
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
..
_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 support for `dim=None` to `AtenMeanDimOp` (#1129) 2022-08-02 16:08:06 +00:00
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py [MHLO] Init MHLO integration. (#1083) 2022-07-20 16:18:16 -07:00
compiler_utils.py Add named exception TorchMlirCompilerError. (#1064) 2022-07-15 16:32:36 -05:00