torch-mlir/python
Maksim Levental 3e999beaea
Small bug fixes in eager mode (#691)
2022-03-28 13:31:07 -05:00
..
test Make error reporting a bit better. 2021-10-28 13:20:16 -07:00
torch_mlir Small bug fixes in eager mode (#691) 2022-03-28 13:31:07 -05:00
torch_mlir_e2e_test Centralize all test serialization logic. 2022-03-28 10:17:13 -07: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
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00