torch-mlir/e2e_testing/torchscript
Prashant Kumar fd505db2c6 Adding support for returning elemental types.
Support for returning elemental types. Previously, only
memref types as returning types was supported. All the hacky ways
to write tests which return elemental types should be taken care of.

Signed-off-by: Prashant Kumar <prashant@nod-labs.com>
2021-11-08 22:20:48 +05:30
..
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
basic.py Adding support for returning elemental types. 2021-11-08 22:20:48 +05:30
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 lowering of aten.floor op 2021-11-06 17:31:44 -04:00
main.py [MLIR][TORCH] Add E2E support for `torch.aten.view` 2021-10-29 22:33:10 +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
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 [tosa] Add Torch.sigmoid fp32 to TOSA (#386) 2021-10-28 10:09:12 -07:00