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/ATen/Transforms
History
Sean Silva 3d08c83580 Add flatten op recognition + shape refinement. 
This op has complex aliasing semantics, so it is kept mutable for now.

With this, we reduce ResNet18 to a single BB with all aten operators
having rank + dtype:
https://gist.github.com/silvasean/2fcb1c6e4d4ae27461204a43ae9c5031
 		2021-05-03 09:54:44 -07:00
..
ATenLayerNamePass.cpp Bump llvm-project to c68d2895a1f4019b387c69d1e5eec31b0eb5e7b0 2021-02-22 12:23:24 -08:00
ATenOpReport.cpp NFC: Transition ATen passes to tablegen registration. 2020-10-22 17:12:44 -07:00
CMakeLists.txt Remove old aten-to-std / ATenLowering pass. 2021-04-30 10:57:02 -07:00
LivenessReport.cpp Bump llvm-project to 444822d77a7fea28aa49edf24533c987efa1b2ee 2020-12-11 14:43:38 -08:00
PassDetail.h NFC: Transition ATen passes to tablegen registration. 2020-10-22 17:12:44 -07:00
Passes.cpp NFC: Transition ATen passes to tablegen registration. 2020-10-22 17:12:44 -07:00
RecognizeKernelsPass.cpp Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00
ReturnEliminationPass.cpp Bump llvm-project to 0524a09cc7e1a0797982feacf505825231efbee7 2021-03-23 14:29:05 -07:00