torch-mlir/lib/Dialect/Torch/Transforms
Vivek Khandelwal 769f3a8870 [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op
This commit adds lowering of `max_pool2d_with_indices` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-04-18 21:05:19 +05:30
..
AdjustCallingConventions.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
CMakeLists.txt Introduce new shape library design. 2022-03-15 12:41:58 -07:00
DecomposeComplexOps.cpp Add 2D case for convolution (#693) 2022-04-08 00:47:57 -04:00
DropShapeCalculations.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
GlobalizeObjectGraph.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
InlineGlobalSlots.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
MaximizeValueSemantics.cpp This commit decomposes aten._reshape_alias op into aten.view op. (#690) 2022-03-28 23:54:28 -05:00
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp Introduce new shape library design. 2022-03-15 12:41:58 -07:00
PrepareForGlobalizeObjectGraph.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
ReduceOpVariants.cpp [LINALG] Add E2E support for `aten.zero_` op 2022-03-25 12:46:50 +05:30
RefinePublicReturn.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
RefineTypes.cpp [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
ReifyShapeCalculations.cpp Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
ShapeLibrary.cpp [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
SimplifyShapeCalculations.cpp Improve list index normalization SimplifyShapeCalculations. (#710) 2022-03-29 22:21:47 +02:00