torch-mlir/test/Dialect/Torch
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
..
GlobalizeObjectGraph Move GlobalizeObjectGraph tests to their own directory 2021-04-27 12:18:54 -07:00
adjust-calling-conventions.mlir Add torch-adjust-calling-conventions pass. 2021-04-05 17:56:35 -07:00
inline-global-slots.mlir Add InlineGlobalSlots pass. 2021-04-27 12:18:54 -07:00
invalid.mlir Add basic MLP's to the e2e curriculum. 2021-04-27 12:18:54 -07:00
ops.mlir Add basic MLP's to the e2e curriculum. 2021-04-27 12:18:54 -07:00
prepare-for-globalize-object-graph.mlir Support multiple instances of a class in GlobalizeObjectGraph. 2021-03-11 19:21:07 -08:00
refine-types.mlir Add flatten op recognition + shape refinement. 2021-05-03 09:54:44 -07:00