torch-mlir/python/torch_mlir/dialects/torch
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
..
importer Add support for `dim=None` to `AtenMeanDimOp` (#1129) 2022-08-02 16:08:06 +00:00
__init__.py Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00