llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch
History
Yi Zhang 732a76f45c Make broadcasting result shape more static 
This involes the following 2 parts:
- Change refine type to propagate more static shape info.
- Get as much static shape info as possible when creating the result
tensor when converting to linalg.
 		2022-01-06 18:39:27 -05:00
..
IR [TORCH][MLIR] Fold trivial cases of `aten.to.dtype` and `aten.view` op 2021-12-24 13:32:34 +05:30
Transforms Make broadcasting result shape more static 2022-01-06 18:39:27 -05:00
Utils Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00