torch-mlir/e2e_testing/torchscript
Ramiro Leal-Cavazos a2392a0f19 Fix bug in handling of pin_memory in AtenOnesOp conversion
This commit fixes a bug with the way ConvertAtenOnesOp was matching on
the pin_memory bool argument, which always resulted in a failed match.
2021-11-12 11:38:25 -05:00
..
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
backprop.py Add e2e testing for aten_tanh_backward op. 2021-11-09 11:28:49 -05:00
basic.py Fix bug in handling of pin_memory in AtenOnesOp conversion 2021-11-12 11:38:25 -05: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 Rsqrt 2021-11-09 11:08:28 -05:00
main.py Add e2e support for aten._softmax_backward_data. 2021-11-09 13:09:30 +05:30
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
quantized_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reduction.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
type_conversion.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
type_promotion.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05: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 [mlir][tosa] Refactor conversions to use templates (#416) 2021-11-11 16:15:58 -08:00