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/torch_mlir/eager_mode
History
Maksim Levental 3e999beaea
Small bug fixes in eager mode (#691) 		2022-03-28 13:31:07 -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 Small bug fixes in eager mode (#691) 2022-03-28 13:31:07 -05:00
torch_mlir_tensor.py Small bug fixes in eager mode (#691) 2022-03-28 13:31:07 -05:00