torch-mlir/lib/Dialect/Torch/Transforms
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
..
AdjustCallingConventions.cpp Add torch-adjust-calling-conventions pass. 2021-04-05 17:56:35 -07:00
CMakeLists.txt Add AllowsTypeRefinement trait and use it to improve RefineTypes 2021-04-30 10:57:02 -07:00
GlobalizeObjectGraph.cpp Bump llvm-project to 484b6648fdd4b104eaf7a2504dd07b60af2c9f8d 2021-04-22 18:12:55 -07:00
InlineGlobalSlots.cpp Add InlineGlobalSlots pass. 2021-04-27 12:18:54 -07:00
PassDetail.h Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00
Passes.cpp Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
PrepareForGlobalizeObjectGraph.cpp Fix issue with unused functions in torch::jit::CompilationUnit 2021-04-20 12:00:35 -07:00
RefineTypes.cpp Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00