torch-mlir/test/Dialect/ATen
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_addmm.mlir Add ATen Dialect (#16) 2020-08-12 19:28:04 -07:00
canonicalize.mlir Add aten::len.t, aten::size, and aten::gt.int primitive ops 2021-04-30 10:57:02 -07:00
recognize_aten_kernels.mlir Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00