torch-mlir/lib/Dialect/Torch/Transforms
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
..
AdjustCallingConventions.cpp Add torch-adjust-calling-conventions pass. 2021-04-05 17:56:35 -07:00
CMakeLists.txt Add AllowsTypeRefinement trait and use it to improve RefineTypes 2021-04-30 10:57:02 -07:00
GlobalizeObjectGraph.cpp Bump llvm-project to 484b6648fdd4b104eaf7a2504dd07b60af2c9f8d 2021-04-22 18:12:55 -07:00
InlineGlobalSlots.cpp Add InlineGlobalSlots pass. 2021-04-27 12:18:54 -07:00
PassDetail.h Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00
Passes.cpp Add recognition/folder/lowering for aten::__is__, aten::ne.int, and aten::dim 2021-04-30 10:57:02 -07:00
PrepareForGlobalizeObjectGraph.cpp Fix issue with unused functions in torch::jit::CompilationUnit 2021-04-20 12:00:35 -07:00
RefineTypes.cpp Add some more ResNet ops. 2021-04-30 10:57:02 -07:00