torch-mlir/lib/Dialect/ATen
Sean Silva ec6d06aa86 Add some more ResNet ops.
- aten::relu_, aten::max_pool2d, aten::adaptive_avg_pool2d, aten::batch_norm, aten::conv2d

No aten-to-linalg conversion for the latter ones, as they are fairly
substantial. At this point, I'm trying to get shape inference and stuff
working for them and the IR cleaned up.
2021-04-30 10:57:02 -07:00
..
IR Add some more ResNet ops. 2021-04-30 10:57:02 -07:00
Transforms Remove old aten-to-std / ATenLowering pass. 2021-04-30 10:57:02 -07:00
CMakeLists.txt NFC: Re-organize ATen directory structure and fix warnings. 2020-10-22 14:13:26 -07:00