llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python
History
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
..
test Fix kwarg annotation in eager (#747) 2022-04-11 17:35:42 -05:00
torch_mlir [MLIR][TORCH] Add E2E support for aten::max_pool2d_with_indices_backward op 2022-04-14 21:46:47 +05:30
torch_mlir_e2e_test [MLIR][TORCH] Add E2E support for aten::max_pool2d_with_indices_backward op 2022-04-14 21:46:47 +05:30
CMakeLists.txt This PR implements an eager mode backend for PyTorch through the torch-mlir framework. This is accomplished by overriding the `__torch_dispatch__` class method on wrapper subclass `TorchMLIRTensor(torch.Tensor)`. 2022-03-22 14:42:57 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00