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 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
..
_torch_mlir_custom_op_example Re-enable custom op support 2022-08-16 22:49:08 +05:30
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc Propagate device data names (#1157) 2022-08-16 09:30:22 -04:00
dialects Clean up shape functions that use `sum_mean_dim` (#1217) 2022-08-18 08:23:43 -07:00
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