torch-mlir/lib/Dialect/ATen/IR
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
..
ATenDialect.cpp Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
ATenDialectOpStats.cpp Add some more ResNet ops. 2021-04-30 10:57:02 -07:00
ATenOps.cpp Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
CMakeLists.txt Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00