Prashant Kumar 977b1b03ea Add aten::nll_loss_forward op lowering. ... The op lowering has been added as a part of `torch-lower-to-linalg` pass. This takes care of ignore_index but the weight and reduction operand is still to be accounted for. Signed-off-by: Prashant Kumar <prashant@nod-labs.com>		2021-12-07 17:11:08 +05:30
..
argmax.py	add argmax lowering	2021-10-13 14:31:16 -04:00
backprop.py	Add e2e test for aten.log_softmax_back_data op	2021-11-19 00:08:28 +05:30
basic.py	Add scalar type promotion for mul and div (#454)	2021-12-03 13:51:25 -06:00
batchnorm.py	E2e support for layernorm.	2021-10-04 14:15:13 -04:00
conv.py	Dual license the torch-mlir project.	2021-10-01 10:46:08 -07:00
elementwise.py	Add scalar type promotion for mul and div (#454)	2021-12-03 13:51:25 -06:00
main.py	Add aten::nll_loss_forward op lowering.	2021-12-07 17:11:08 +05:30
matmul.py	Add lowering of aten.matmul op.	2021-10-26 12:45:09 -04:00
mlp.py	Support aten::linear with rank 3 inputs	2021-11-18 22:15:04 +05:30
nll_loss.py	Add aten::nll_loss_forward op lowering.	2021-12-07 17:11:08 +05:30
quantized_models.py	Dual license the torch-mlir project.	2021-10-01 10:46:08 -07:00
reduction.py	[MLIR][TORCH] Add E2E support for aten.mean and aten.numel op.	2021-12-02 11:51:13 +05:30
scalar.py	add aten.add.int lowering in TorchToStd	2021-11-29 13:22:50 -05:00
slice_like.py	Add lowering for slice and selectInt (#398)	2021-12-02 22:09:21 -06:00
squeeze.py	[MLIR][TORCH] Add E2E support for `aten.squeeze` op	2021-11-30 23:00:28 +05:30
type_conversion.py	Add convertScalarToDtype helper.	2021-11-08 17:50:52 -05:00
type_promotion.py	Add scalar type promotion for mul and div (#454)	2021-12-03 13:51:25 -06:00
view.py	[MLIR][TORCH] Add E2E support for `torch.aten.view`	2021-10-29 22:33:10 +05:30
vision_models.py	Add min/max/clamp support.	2021-10-27 13:29:21 -07:00
xfail_sets.py	Update external llvm to 966b72098363d44adf2882b9c34	2021-12-06 23:33:58 +05:30