torch-mlir/python/torch_mlir/eager_mode
Maksim Levental d46f169c1a
Fix kwarg annotation in eager (#747)
2022-04-11 17:35:42 -05:00
..
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
__init__.py 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
ir_building.py Small bug fixes in eager mode (#691) 2022-03-28 13:31:07 -05:00
torch_mlir_dispatch.py Fix kwarg annotation in eager (#747) 2022-04-11 17:35:42 -05:00
torch_mlir_tensor.py Add link to forum post describing `__torch_dispatch__` 2022-04-01 10:10:43 -07:00