torch-mlir/python
Ramiro Leal-Cavazos f07f7d20f9
Clean up shape functions that use `sum_mean_dim` (#1217)
I recently fixed the handling of the `dim` argument in
`sum_mean_dim` (59fccab857). Therefore,
the checks that the `dim` input is `None` or `[]` are no longer needed.
2022-08-18 08:23:43 -07:00
..
test Propagate device data names (#1157) 2022-08-16 09:30:22 -04:00
torch_mlir Clean up shape functions that use `sum_mean_dim` (#1217) 2022-08-18 08:23:43 -07:00
torch_mlir_e2e_test Re-enable custom op support 2022-08-16 22:49:08 +05:30
CMakeLists.txt Re-enable custom op support 2022-08-16 22:49:08 +05:30
TorchMLIRModule.cpp Revert "python: trim registration and loading of dialects and passes" (#1093) 2022-07-21 09:35:42 -07:00