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
..
test	Fix kwarg annotation in eager (#747)	2022-04-11 17:35:42 -05:00
torch_mlir	[MLIR][TORCH] Add E2E support for max_pool2d_with_indices op	2022-04-18 21:05:19 +05:30
torch_mlir_e2e_test	[MLIR][TORCH] Add E2E support for max_pool2d_with_indices op	2022-04-18 21:05:19 +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