torch-mlir/lib/Dialect/Torch
Vivek Khandelwal 1bccb4fc8a [MLIR][TORCH] Add E2E support for aten::max_pool2d_with_indices_backward op
This commit adds lowering of `aten::max_pool2d_with_indices_backward` op.

This commit also fixes formatting issues in basic.py.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-04-14 21:46:47 +05:30
..
IR Fix for 0-size dim inferred incorrectly. 2022-03-30 16:36:15 -07:00
Transforms [MLIR][TORCH] Add E2E support for aten::max_pool2d_with_indices_backward op 2022-04-14 21:46:47 +05:30
Utils Fixes https://github.com/llvm/torch-mlir/issues/751 where `torch.bool` is parsed as signless `i1`. (#752) 2022-04-13 12:28:27 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00