torch-mlir/lib/Conversion
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
..
TorchToLinalg [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
TorchToSCF Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad 2022-03-18 13:16:14 -04:00
TorchToStd [LINALG] Add decomposition of `aten.dropout` op 2022-03-22 13:14:49 +05:30
TorchToTMTensor [MLIR][TORCH] Add E2E support for aten::max_pool2d_with_indices_backward op 2022-04-14 21:46:47 +05:30
TorchToTosa Add 2D case for convolution (#693) 2022-04-08 00:47:57 -04:00
Utils [MLIR][TORCH] Add E2E support for max_pool2d_with_indices op 2022-04-18 21:05:19 +05:30
CMakeLists.txt [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30
PassDetail.h Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
Passes.cpp [MLIR][TORCH] Add TorchToTMTensor pass 2022-03-08 22:52:34 +05:30