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
..
CAPI	Add minimal support for Union types.	2022-03-29 17:45:48 -07:00
Conversion	Add 1D, weight, and reduction support to nll_loss_backward (#729)	2022-04-04 10:57:49 -07:00
Dialect	Add support for multi-dim input to `index_put_impl` (#722)	2022-03-31 09:27:21 -07:00
RefBackend	Bump LLVM at 8361c5da30588d3d4a48eae648f53be1feb5cfad	2022-03-18 13:16:14 -04:00
CMakeLists.txt	Add TMTensor dialect to torch-mlir	2022-02-15 16:45:38 -05:00
InitAll.cpp	Add TMTensor dialect to torch-mlir	2022-02-15 16:45:38 -05:00