torch-mlir/lib
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
..
Backend Miscellaneous changes while trying to work on ResNet18 2021-04-27 11:51:11 -07:00
CAPI Add npcomp-verify-backend-contract pass. 2021-04-20 12:00:35 -07:00
Conversion Add aten::len.t, aten::size, and aten::gt.int primitive ops 2021-04-30 10:57:02 -07:00
Dialect Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00
Interfaces Add AllowsTypeRefinement trait and use it to improve RefineTypes 2021-04-30 10:57:02 -07:00
RefBackend Bump llvm-project to 484b6648fdd4b104eaf7a2504dd07b60af2c9f8d 2021-04-22 18:12:55 -07:00
Typing Bump llvm-project to c68d2895a1f4019b387c69d1e5eec31b0eb5e7b0 2021-02-22 12:23:24 -08:00
CMakeLists.txt Add AllowsTypeRefinement trait and use it to improve RefineTypes 2021-04-30 10:57:02 -07:00
InitAll.cpp Add npcomp-verify-backend-contract pass. 2021-04-20 12:00:35 -07:00