Ramiro Leal-Cavazos 5620fe030e Add 1D, weight, and reduction support to nll_loss_backward (#729) ... This commit adds the following support to the op `nll_loss_backward`: - `input` tensor can be rank-1 - `weight` parameter - `reduction` parameter - `target`, `grad_output`, `total_weight` can be rank-0 - Checks that input tensors are of the expected type		2022-04-04 10:57:49 -07:00
..
test	Make error reporting a bit better.	2021-10-28 13:20:16 -07:00
torch_mlir	Add link to forum post describing `__torch_dispatch__`	2022-04-01 10:10:43 -07:00
torch_mlir_e2e_test	Add 1D, weight, and reduction support to nll_loss_backward (#729)	2022-04-04 10:57:49 -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