torch-mlir/test/Dialect
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
..
ATen Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00
Basicpy Constant fold through basicpy.bool_cast. 2021-04-30 10:57:02 -07:00
Numpy Add RefinePublicReturn pass. 2021-04-07 11:06:34 -07:00
Refback [RefBackend] Use std.global_memref instead of homegrown thing 2020-11-13 18:43:50 -08:00
Refbackrt [refbackrt] Scalar arg support 2021-03-23 13:16:44 -07:00
TCF Add TCF convolutional op with bias addition (#137) 2020-12-15 12:53:12 -08:00
TCP Bump llvm-project to 0524a09cc7e1a0797982feacf505825231efbee7 2021-03-23 14:29:05 -07:00
Torch Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00