torch-mlir/lib/Dialect/Torch/Transforms
Yi Zhang 0342b73bf1 Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering
- torch.aten.flatten.using_ints to linalg lowering
- torch.aten.max_pool2d to linalg lowering
- Support torch.aten.conv2d for more flexible dilation and strides values
2021-08-04 12:00:43 -04:00
..
AdjustCallingConventions.cpp Make MaximizeValueSemantics a bit smarter. 2021-06-22 16:48:57 -07:00
BackendTypeConversion.cpp Add a !torch.float type. 2021-06-17 09:24:18 -07:00
CMakeLists.txt Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00
GlobalizeObjectGraph.cpp Add a !torch.float type. 2021-06-17 09:24:18 -07:00
InlineGlobalSlots.cpp Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
MaximizeValueSemantics.cpp Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00
PassDetail.h Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00
Passes.cpp Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
PrepareForGlobalizeObjectGraph.cpp Add `!torch.bool` type. 2021-06-16 13:22:00 -07:00
ReduceOpVariants.cpp Make MaximizeValueSemantics a bit smarter. 2021-06-22 16:48:57 -07:00
RefinePublicReturn.cpp Introduce `!torch.tensor` / `!torch.vtensor` types. 2021-06-10 10:56:48 -07:00
RefineTypes.cpp Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering 2021-08-04 12:00:43 -04:00
VerifyInvariantsBeforeBackendLowering.cpp Generalize support for elementwise ops. 2021-06-28 13:28:38 -07:00