torch-mlir/python
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
..
test Clean up Autogen (#1112) 2022-07-30 09:40:02 -04:00
torch_mlir Add support for `dim=None` to `AtenMeanDimOp` (#1129) 2022-08-02 16:08:06 +00:00
torch_mlir_e2e_test Add support for `dim=None` to `AtenMeanDimOp` (#1129) 2022-08-02 16:08:06 +00:00
CMakeLists.txt Resolve remaining LTC CI failures (#1110) 2022-07-30 09:40:02 -04:00
TorchMLIRModule.cpp Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00